┏━━━━┳━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┳━━━━━━━━┓
┃ ┃ Name ┃ Type ┃ Params ┃
┡━━━━╇━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━╇━━━━━━━━┩
│ 0 │ criterion │ WeightedMSELoss │ 0 │
│ 1 │ model │ PFTemporalFusionTransformerModel │ 1.3 M │
│ 2 │ model.static_vs │ VariableSelection │ 39.0 K │
│ 3 │ model.enc_vs │ VariableSelection │ 117 K │
│ 4 │ model.dec_vs │ VariableSelection │ 78.4 K │
│ 5 │ model.static_ctxt_vs │ GatedResidualNetwork │ 66.3 K │
│ 6 │ model.static_ctxt_enrichment │ GatedResidualNetwork │ 66.3 K │
│ 7 │ model.lstm_init_hidden │ GatedResidualNetwork │ 66.3 K │
│ 8 │ model.lstm_init_cell │ GatedResidualNetwork │ 66.3 K │
│ 9 │ model.enc_lstm │ LSTM │ 264 K │
│ 10 │ model.dec_lstm │ LSTM │ 264 K │
│ 11 │ model.enc_gate1 │ GatedLinearUnit │ 33.0 K │
│ 12 │ model.enc_norm1 │ AddNorm │ 256 │
│ 13 │ model.static_enrichment │ GatedResidualNetwork │ 82.7 K │
│ 14 │ model.attn │ InterpretableMultiHeadAttention │ 41.4 K │
│ 15 │ model.attn_gate1 │ GatedLinearUnit │ 33.0 K │
│ 16 │ model.attn_norm1 │ AddNorm │ 256 │
│ 17 │ model.attn_grn │ GatedResidualNetwork │ 66.3 K │
│ 18 │ model.attn_gate2 │ GatedLinearUnit │ 33.0 K │
│ 19 │ model.attn_norm2 │ AddNorm │ 256 │
│ 20 │ model.output_layer │ Linear │ 129 │
└────┴──────────────────────────────┴──────────────────────────────────┴────────┘
https://app.neptune.ai/o/hysteresis/org/TFT/runs/details?viewId=standard-view&shortId=TFT-6&type=run
Bdot only input