| { | |
| "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" | |
| } |