A pytorch implementation of Eclipse
Project description
Eclipse
Implementing Paletta et al. in Pytorch
Install
pip install eclipse_pytorch
How to use
import torch
from eclipse_pytorch.model import *
eclipse = Eclipse()
preds = eclipse([torch.rand(2, 3, 256, 256) for _ in range(4)])
states.shape=torch.Size([2, 4, 128, 32, 32])
present_state.shape=torch.Size([2, 1, 128, 32, 32])
hidden_state.shape=torch.Size([2, 128, 32, 32])
future_states.shape=torch.Size([2, 6, 128, 32, 32])
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
eclipse_pytorch-0.0.1.tar.gz
(15.2 kB
view hashes)
Built Distribution
Close
Hashes for eclipse_pytorch-0.0.1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 1e057bf3764d6e5a984223bcaf7920f71f18880bede7c519c5fdbe12dff2a85f |
|
MD5 | 532fd90a364d2c3930274f551a20ab27 |
|
BLAKE2b-256 | d9e9756a0f018954189596bee23ca901d33247a79a00c71d43e12b3ad5e4325c |