CURV-1

┏━━━┳━━━━━━━━━━━━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━━━━━━━━┳━━━━━━━━┓
┃   ┃ Name                     ┃ Type                 ┃ Params ┃
┡━━━╇━━━━━━━━━━━━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━━━━━━━━╇━━━━━━━━┩
│ 0 │ criterion                │ MSELoss              │      0 │
│ 1 │ model                    │ TransformerLSTMModel │  4.9 M │
│ 2 │ model.encoder_grn        │ GatedResidualNetwork │  200 K │
│ 3 │ model.decoder_grn        │ GatedResidualNetwork │  199 K │
│ 4 │ model.encoder            │ LSTM                 │  1.1 M │
│ 5 │ model.decoder            │ LSTM                 │  1.1 M │
│ 6 │ model.transformer_blocks │ ModuleList           │  2.4 M │
│ 7 │ model.output_head        │ Linear               │    257 │
└───┴──────────────────────────┴──────────────────────┴────────┘
Trainable params: 4.9 M
Non-trainable params: 0
Total params: 4.9 M
Total estimated model params size (MB): 19

TransformerLSTM in first stage of Curriculum Learning for Learn model to use voltage.