{ "best_metric": 0.8254, "best_model_checkpoint": "imagenet-tiger/checkpoint-3750", "epoch": 3.0, "eval_steps": 500, "global_step": 3750, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.08, "grad_norm": 11.370553970336914, "learning_rate": 9.982464296247522e-05, "loss": 1.1275, "step": 100 }, { "epoch": 0.16, "grad_norm": 11.340897560119629, "learning_rate": 9.929980185352526e-05, "loss": 1.1066, "step": 200 }, { "epoch": 0.24, "grad_norm": 9.576211929321289, "learning_rate": 9.842915805643155e-05, "loss": 1.1377, "step": 300 }, { "epoch": 0.32, "grad_norm": 10.26335334777832, "learning_rate": 9.721881851187406e-05, "loss": 1.1071, "step": 400 }, { "epoch": 0.4, "grad_norm": 9.477747917175293, "learning_rate": 9.567727288213005e-05, "loss": 1.1081, "step": 500 }, { "epoch": 0.48, "grad_norm": 7.301119804382324, "learning_rate": 9.381533400219318e-05, "loss": 1.0873, "step": 600 }, { "epoch": 0.56, "grad_norm": 11.125399589538574, "learning_rate": 9.164606203550497e-05, "loss": 1.1031, "step": 700 }, { "epoch": 0.64, "grad_norm": 13.228848457336426, "learning_rate": 8.9184672866292e-05, "loss": 1.0457, "step": 800 }, { "epoch": 0.72, "grad_norm": 7.878283500671387, "learning_rate": 8.644843137107059e-05, "loss": 1.1162, "step": 900 }, { "epoch": 0.8, "grad_norm": 5.097326755523682, "learning_rate": 8.345653031794292e-05, "loss": 1.1242, "step": 1000 }, { "epoch": 0.88, "grad_norm": 12.475173950195312, "learning_rate": 8.022995574311876e-05, "loss": 1.1094, "step": 1100 }, { "epoch": 0.96, "grad_norm": 12.513256072998047, "learning_rate": 7.679133974894983e-05, "loss": 1.0771, "step": 1200 }, { "epoch": 1.0, "eval_accuracy": 0.7971, "eval_loss": 0.7663370370864868, "eval_runtime": 202.227, "eval_samples_per_second": 49.449, "eval_steps_per_second": 1.548, "step": 1250 }, { "epoch": 1.04, "grad_norm": 10.96892261505127, "learning_rate": 7.316480175599309e-05, "loss": 1.1045, "step": 1300 }, { "epoch": 1.12, "grad_norm": 12.19508171081543, "learning_rate": 6.937577932260515e-05, "loss": 0.9819, "step": 1400 }, { "epoch": 1.2, "grad_norm": 16.93886375427246, "learning_rate": 6.545084971874738e-05, "loss": 0.967, "step": 1500 }, { "epoch": 1.28, "grad_norm": 12.414902687072754, "learning_rate": 6.141754350553279e-05, "loss": 0.9287, "step": 1600 }, { "epoch": 1.3599999999999999, "grad_norm": 16.11931610107422, "learning_rate": 5.730415142812059e-05, "loss": 0.8632, "step": 1700 }, { "epoch": 1.44, "grad_norm": 10.42251968383789, "learning_rate": 5.313952597646568e-05, "loss": 0.9125, "step": 1800 }, { "epoch": 1.52, "grad_norm": 11.504471778869629, "learning_rate": 4.895287900583216e-05, "loss": 0.9045, "step": 1900 }, { "epoch": 1.6, "grad_norm": 9.115998268127441, "learning_rate": 4.477357683661734e-05, "loss": 0.8667, "step": 2000 }, { "epoch": 1.6800000000000002, "grad_norm": 10.520715713500977, "learning_rate": 4.063093427071376e-05, "loss": 0.8806, "step": 2100 }, { "epoch": 1.76, "grad_norm": 12.112250328063965, "learning_rate": 3.655400896923672e-05, "loss": 0.884, "step": 2200 }, { "epoch": 1.8399999999999999, "grad_norm": 12.806370735168457, "learning_rate": 3.257139763390925e-05, "loss": 0.807, "step": 2300 }, { "epoch": 1.92, "grad_norm": 12.071837425231934, "learning_rate": 2.8711035421746367e-05, "loss": 0.8634, "step": 2400 }, { "epoch": 2.0, "grad_norm": 9.2099609375, "learning_rate": 2.500000000000001e-05, "loss": 0.8206, "step": 2500 }, { "epoch": 2.0, "eval_accuracy": 0.8207, "eval_loss": 0.6771770119667053, "eval_runtime": 200.5917, "eval_samples_per_second": 49.853, "eval_steps_per_second": 1.56, "step": 2500 }, { "epoch": 2.08, "grad_norm": 9.109099388122559, "learning_rate": 2.1464321615778422e-05, "loss": 0.7565, "step": 2600 }, { "epoch": 2.16, "grad_norm": 7.973142623901367, "learning_rate": 1.8128800512565513e-05, "loss": 0.7475, "step": 2700 }, { "epoch": 2.24, "grad_norm": 13.218464851379395, "learning_rate": 1.5016832974331724e-05, "loss": 0.7201, "step": 2800 }, { "epoch": 2.32, "grad_norm": 7.9871134757995605, "learning_rate": 1.2150247217412186e-05, "loss": 0.7449, "step": 2900 }, { "epoch": 2.4, "grad_norm": 7.4544291496276855, "learning_rate": 9.549150281252633e-06, "loss": 0.7013, "step": 3000 }, { "epoch": 2.48, "grad_norm": 8.663176536560059, "learning_rate": 7.2317869919746705e-06, "loss": 0.7347, "step": 3100 }, { "epoch": 2.56, "grad_norm": 12.630393028259277, "learning_rate": 5.214411988029355e-06, "loss": 0.7542, "step": 3200 }, { "epoch": 2.64, "grad_norm": 8.864751815795898, "learning_rate": 3.511175705587433e-06, "loss": 0.6819, "step": 3300 }, { "epoch": 2.7199999999999998, "grad_norm": 6.5524725914001465, "learning_rate": 2.134025123396638e-06, "loss": 0.7919, "step": 3400 }, { "epoch": 2.8, "grad_norm": 7.316872596740723, "learning_rate": 1.0926199633097157e-06, "loss": 0.6864, "step": 3500 }, { "epoch": 2.88, "grad_norm": 9.99939250946045, "learning_rate": 3.9426493427611177e-07, "loss": 0.7364, "step": 3600 }, { "epoch": 2.96, "grad_norm": 12.940382957458496, "learning_rate": 4.385849505708084e-08, "loss": 0.7212, "step": 3700 }, { "epoch": 3.0, "eval_accuracy": 0.8254, "eval_loss": 0.6595176458358765, "eval_runtime": 198.5571, "eval_samples_per_second": 50.363, "eval_steps_per_second": 1.576, "step": 3750 }, { "epoch": 3.0, "step": 3750, "total_flos": 9.37824189087744e+18, "train_loss": 0.9114908182779948, "train_runtime": 5966.5057, "train_samples_per_second": 20.112, "train_steps_per_second": 0.629 } ], "logging_steps": 100, "max_steps": 3750, "num_input_tokens_seen": 0, "num_train_epochs": 3, "save_steps": 500, "total_flos": 9.37824189087744e+18, "train_batch_size": 32, "trial_name": null, "trial_params": null }