{ "device": "cuda", "steps": 300, "results": { "sgd": { "loss_checkpoints": [ 0.6759916543960571, 0.6073195934295654, 0.5333767533302307, 0.4328407645225525, 0.3397914171218872, 0.2723374664783478 ], "final_loss": 0.22558128833770752, "train_accuracy": 0.943359375, "seconds": 0.6605247830002554 }, "momentum": { "loss_checkpoints": [ 0.6759916543960571, 0.38542139530181885, 0.15265634655952454, 0.09576870501041412, 0.07248643040657043, 0.05970747768878937 ], "final_loss": 0.051639359444379807, "train_accuracy": 0.99169921875, "seconds": 0.5288700599994627 }, "third_order": { "loss_checkpoints": [ 0.6759916543960571, 0.6549453139305115, 0.6521938443183899, 0.6337970495223999, 0.6934977173805237, 1.4368056058883667 ], "final_loss": 1.7160285711288452, "train_accuracy": 0.51318359375, "seconds": 1.029801511000187 } } }