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.2.tar.gz
(14.4 kB
view hashes)
Built Distribution
Close
Hashes for eclipse_pytorch-0.0.2-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 929f99ca0c0d050e405386c94d767bd17f64e2ea30dc663c3eb365d0478cf101 |
|
MD5 | 1d443a18c7db987e2c39554dac0528df |
|
BLAKE2b-256 | bd0098edce66aa018d82a02b363ccdde7b9af62965f233af13acbc9ad985ebc8 |