Cyclic Lie-Bracket Residual Block / toy_results.json
Beats tuned baseline
1{
2 "device": "cuda",
3 "baseline": {
4 "train_mse": 0.00021877672406844795,
5 "test_mse": 0.00022434978745877743,
6 "loss_step_50": 0.07765127718448639,
7 "loss_step_350": 0.00022028385137673467,
8 "seconds": 1.6991870403289795,
9 "parameters": 1170
10 },
11 "cyclic": {
12 "train_mse": 0.0028530012350529432,
13 "test_mse": 0.003148798132315278,
14 "loss_step_50": 0.0983579009771347,
15 "loss_step_350": 0.002886241301894188,
16 "seconds": 3.4385135173797607,
17 "parameters": 2274,
18 "order_signal": 4.741206066682935e-05
19 }
20}