artifacts/checkpoints/model_config.json

{
  "input_dim": 27,
  "hidden_dim": 256,
  "num_blocks": 4,
  "dropout": 0.1,
  "num_classes": 3,
  "heteroscedastic": true
}