config.json

{
  "window_size": 90,
  "action_dim": 6,
  "hidden_dim": 512,
  "bottleneck_dim": 8,
  "dropout": 0.1,
  "mean": [
    5.762328624725342,
    133.8108673095703,
    132.24156188964844,
    63.754695892333984,
    87.85747528076172,
    8.232730865478516
  ],
  "std": [
    25.211471557617188,
    41.20315170288086,
    32.92947769165039,
    14.849191665649414,
    12.994131088256836,
    9.438411712646484
  ],
  "learning_rate": 0.0017612312418199766,
  "batch_size": 32,
  "epochs": 160,
  "train_stride": 4,
  "device": "cuda"
}