iota2.learning.pytorch

Modules

iota2.learning.pytorch.lightweight_temporal_attention_encoder

Lightweight Temporal Attention Encoder module

iota2.learning.pytorch.torch_nn_bank

Module containing all natively usable pytorch neural networks in iota2