{ "best_metric": 0.16110308468341827, "best_model_checkpoint": "/robodata/smodak/Projects/nspl/scripts/terrainseg/training/models/dropoff-utcustom-train-SF-RGB-b0_7/checkpoint-240", "epoch": 80.0, "global_step": 240, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.33, "learning_rate": 5e-06, "loss": 1.2178, "step": 1 }, { "epoch": 0.67, "learning_rate": 1e-05, "loss": 1.2142, "step": 2 }, { "epoch": 1.0, "learning_rate": 1.5e-05, "loss": 1.2223, "step": 3 }, { "epoch": 1.33, "learning_rate": 2e-05, "loss": 1.208, "step": 4 }, { "epoch": 1.67, "learning_rate": 2.5e-05, "loss": 1.2072, "step": 5 }, { "epoch": 2.0, "learning_rate": 3e-05, "loss": 1.2015, "step": 6 }, { "epoch": 2.33, "learning_rate": 3.5000000000000004e-05, "loss": 1.1888, "step": 7 }, { "epoch": 2.67, "learning_rate": 4e-05, "loss": 1.1768, "step": 8 }, { "epoch": 3.0, "learning_rate": 4.5e-05, "loss": 1.1543, "step": 9 }, { "epoch": 3.33, "learning_rate": 5e-05, "loss": 1.1505, "step": 10 }, { "epoch": 3.33, "eval_accuracy_dropoff": 0.9456199600248122, "eval_accuracy_undropoff": 0.2616499186280552, "eval_accuracy_unlabeled": NaN, "eval_iou_dropoff": 0.07032648592680048, "eval_iou_undropoff": 0.2615922132145178, "eval_iou_unlabeled": 0.0, "eval_loss": 1.1103304624557495, "eval_mean_accuracy": 0.6036349393264338, "eval_mean_iou": 0.11063956638043942, "eval_overall_accuracy": 0.2919347127278646, "eval_runtime": 1.7484, "eval_samples_per_second": 8.579, "eval_steps_per_second": 0.572, "step": 10 }, { "epoch": 3.67, "learning_rate": 5.500000000000001e-05, "loss": 1.1405, "step": 11 }, { "epoch": 4.0, "learning_rate": 6e-05, "loss": 1.1039, "step": 12 }, { "epoch": 4.33, "learning_rate": 6.500000000000001e-05, "loss": 1.1211, "step": 13 }, { "epoch": 4.67, "learning_rate": 7.000000000000001e-05, "loss": 1.0914, "step": 14 }, { "epoch": 5.0, "learning_rate": 7.500000000000001e-05, "loss": 1.0583, "step": 15 }, { "epoch": 5.33, "learning_rate": 8e-05, "loss": 1.0527, "step": 16 }, { "epoch": 5.67, "learning_rate": 8.5e-05, "loss": 1.0227, "step": 17 }, { "epoch": 6.0, "learning_rate": 9e-05, "loss": 1.1078, "step": 18 }, { "epoch": 6.33, "learning_rate": 8.973684210526316e-05, "loss": 1.0001, "step": 19 }, { "epoch": 6.67, "learning_rate": 8.947368421052632e-05, "loss": 0.9635, "step": 20 }, { "epoch": 6.67, "eval_accuracy_dropoff": 0.8177051025800078, "eval_accuracy_undropoff": 0.8763050644323176, "eval_accuracy_unlabeled": NaN, "eval_iou_dropoff": 0.2434656953279978, "eval_iou_undropoff": 0.8694137027963167, "eval_iou_unlabeled": 0.0, "eval_loss": 1.0114336013793945, "eval_mean_accuracy": 0.8470050835061627, "eval_mean_iou": 0.3709597993747715, "eval_overall_accuracy": 0.8737103780110677, "eval_runtime": 1.7139, "eval_samples_per_second": 8.752, "eval_steps_per_second": 0.583, "step": 20 }, { "epoch": 7.0, "learning_rate": 8.921052631578948e-05, "loss": 1.0405, "step": 21 }, { "epoch": 7.33, "learning_rate": 8.894736842105263e-05, "loss": 0.9095, "step": 22 }, { "epoch": 7.67, "learning_rate": 8.86842105263158e-05, "loss": 0.9104, "step": 23 }, { "epoch": 8.0, "learning_rate": 8.842105263157894e-05, "loss": 0.8736, "step": 24 }, { "epoch": 8.33, "learning_rate": 8.815789473684211e-05, "loss": 0.8849, "step": 25 }, { "epoch": 8.67, "learning_rate": 8.789473684210527e-05, "loss": 0.8143, "step": 26 }, { "epoch": 9.0, "learning_rate": 8.763157894736843e-05, "loss": 0.7846, "step": 27 }, { "epoch": 9.33, "learning_rate": 8.736842105263158e-05, "loss": 0.7748, "step": 28 }, { "epoch": 9.67, "learning_rate": 8.710526315789474e-05, "loss": 0.7513, "step": 29 }, { "epoch": 10.0, "learning_rate": 8.68421052631579e-05, "loss": 0.9358, "step": 30 }, { "epoch": 10.0, "eval_accuracy_dropoff": 0.5848209157534404, "eval_accuracy_undropoff": 0.9605968730608304, "eval_accuracy_unlabeled": NaN, "eval_iou_dropoff": 0.3193744373731639, "eval_iou_undropoff": 0.9425019881457949, "eval_iou_unlabeled": 0.0, "eval_loss": 0.8242360353469849, "eval_mean_accuracy": 0.7727088944071354, "eval_mean_iou": 0.42062547517298626, "eval_overall_accuracy": 0.9439582824707031, "eval_runtime": 1.7189, "eval_samples_per_second": 8.726, "eval_steps_per_second": 0.582, "step": 30 }, { "epoch": 10.33, "learning_rate": 8.657894736842107e-05, "loss": 0.7191, "step": 31 }, { "epoch": 10.67, "learning_rate": 8.631578947368421e-05, "loss": 0.73, "step": 32 }, { "epoch": 11.0, "learning_rate": 8.605263157894737e-05, "loss": 0.6168, "step": 33 }, { "epoch": 11.33, "learning_rate": 8.578947368421054e-05, "loss": 0.6559, "step": 34 }, { "epoch": 11.67, "learning_rate": 8.552631578947369e-05, "loss": 0.6277, "step": 35 }, { "epoch": 12.0, "learning_rate": 8.526315789473684e-05, "loss": 0.6168, "step": 36 }, { "epoch": 12.33, "learning_rate": 8.5e-05, "loss": 0.6095, "step": 37 }, { "epoch": 12.67, "learning_rate": 8.473684210526316e-05, "loss": 0.5927, "step": 38 }, { "epoch": 13.0, "learning_rate": 8.447368421052632e-05, "loss": 0.8162, "step": 39 }, { "epoch": 13.33, "learning_rate": 8.421052631578948e-05, "loss": 0.579, "step": 40 }, { "epoch": 13.33, "eval_accuracy_dropoff": 0.5401589817814231, "eval_accuracy_undropoff": 0.982923599779886, "eval_accuracy_unlabeled": NaN, "eval_iou_dropoff": 0.3951496002117638, "eval_iou_undropoff": 0.9624271055937778, "eval_iou_unlabeled": 0.0, "eval_loss": 0.5702608823776245, "eval_mean_accuracy": 0.7615412907806545, "eval_mean_iou": 0.4525255686018472, "eval_overall_accuracy": 0.9633188883463542, "eval_runtime": 1.8153, "eval_samples_per_second": 8.263, "eval_steps_per_second": 0.551, "step": 40 }, { "epoch": 13.67, "learning_rate": 8.394736842105263e-05, "loss": 0.5659, "step": 41 }, { "epoch": 14.0, "learning_rate": 8.368421052631579e-05, "loss": 0.5447, "step": 42 }, { "epoch": 14.33, "learning_rate": 8.342105263157896e-05, "loss": 0.5265, "step": 43 }, { "epoch": 14.67, "learning_rate": 8.31578947368421e-05, "loss": 0.5263, "step": 44 }, { "epoch": 15.0, "learning_rate": 8.289473684210527e-05, "loss": 0.4922, "step": 45 }, { "epoch": 15.33, "learning_rate": 8.263157894736843e-05, "loss": 0.5099, "step": 46 }, { "epoch": 15.67, "learning_rate": 8.236842105263158e-05, "loss": 0.4706, "step": 47 }, { "epoch": 16.0, "learning_rate": 8.210526315789474e-05, "loss": 0.4809, "step": 48 }, { "epoch": 16.33, "learning_rate": 8.18421052631579e-05, "loss": 0.4888, "step": 49 }, { "epoch": 16.67, "learning_rate": 8.157894736842105e-05, "loss": 0.4411, "step": 50 }, { "epoch": 16.67, "eval_accuracy_dropoff": 0.48720334505019874, "eval_accuracy_undropoff": 0.9888639114094216, "eval_accuracy_unlabeled": NaN, "eval_iou_dropoff": 0.39281844198905264, "eval_iou_undropoff": 0.9659161382921264, "eval_iou_unlabeled": 0.0, "eval_loss": 0.4165554940700531, "eval_mean_accuracy": 0.7380336282298102, "eval_mean_iou": 0.452911526760393, "eval_overall_accuracy": 0.9666514078776042, "eval_runtime": 1.7457, "eval_samples_per_second": 8.593, "eval_steps_per_second": 0.573, "step": 50 }, { "epoch": 17.0, "learning_rate": 8.131578947368422e-05, "loss": 0.4717, "step": 51 }, { "epoch": 17.33, "learning_rate": 8.105263157894737e-05, "loss": 0.4523, "step": 52 }, { "epoch": 17.67, "learning_rate": 8.078947368421052e-05, "loss": 0.4215, "step": 53 }, { "epoch": 18.0, "learning_rate": 8.05263157894737e-05, "loss": 0.4007, "step": 54 }, { "epoch": 18.33, "learning_rate": 8.026315789473685e-05, "loss": 0.4253, "step": 55 }, { "epoch": 18.67, "learning_rate": 8e-05, "loss": 0.4153, "step": 56 }, { "epoch": 19.0, "learning_rate": 7.973684210526316e-05, "loss": 0.6574, "step": 57 }, { "epoch": 19.33, "learning_rate": 7.947368421052632e-05, "loss": 0.3899, "step": 58 }, { "epoch": 19.67, "learning_rate": 7.921052631578948e-05, "loss": 0.4108, "step": 59 }, { "epoch": 20.0, "learning_rate": 7.894736842105263e-05, "loss": 0.4311, "step": 60 }, { "epoch": 20.0, "eval_accuracy_dropoff": 0.44003721827831, "eval_accuracy_undropoff": 0.9911496700950386, "eval_accuracy_unlabeled": NaN, "eval_iou_dropoff": 0.3694591258053316, "eval_iou_undropoff": 0.966086774739557, "eval_iou_unlabeled": NaN, "eval_loss": 0.38426053524017334, "eval_mean_accuracy": 0.7155934441866743, "eval_mean_iou": 0.6677729502724443, "eval_overall_accuracy": 0.966747538248698, "eval_runtime": 1.8089, "eval_samples_per_second": 8.292, "eval_steps_per_second": 0.553, "step": 60 }, { "epoch": 20.33, "learning_rate": 7.868421052631579e-05, "loss": 0.3803, "step": 61 }, { "epoch": 20.67, "learning_rate": 7.842105263157896e-05, "loss": 0.3828, "step": 62 }, { "epoch": 21.0, "learning_rate": 7.815789473684212e-05, "loss": 0.5699, "step": 63 }, { "epoch": 21.33, "learning_rate": 7.789473684210526e-05, "loss": 0.3803, "step": 64 }, { "epoch": 21.67, "learning_rate": 7.763157894736843e-05, "loss": 0.3709, "step": 65 }, { "epoch": 22.0, "learning_rate": 7.736842105263159e-05, "loss": 0.4738, "step": 66 }, { "epoch": 22.33, "learning_rate": 7.710526315789474e-05, "loss": 0.3891, "step": 67 }, { "epoch": 22.67, "learning_rate": 7.68421052631579e-05, "loss": 0.3576, "step": 68 }, { "epoch": 23.0, "learning_rate": 7.657894736842105e-05, "loss": 0.5494, "step": 69 }, { "epoch": 23.33, "learning_rate": 7.631578947368421e-05, "loss": 0.3437, "step": 70 }, { "epoch": 23.33, "eval_accuracy_dropoff": 0.39948767431709054, "eval_accuracy_undropoff": 0.9917728653036201, "eval_accuracy_unlabeled": NaN, "eval_iou_dropoff": 0.33924477871099273, "eval_iou_undropoff": 0.9649275553674269, "eval_iou_unlabeled": 0.0, "eval_loss": 0.35898175835609436, "eval_mean_accuracy": 0.6956302698103554, "eval_mean_iou": 0.43472411135947325, "eval_overall_accuracy": 0.9655476888020833, "eval_runtime": 1.8305, "eval_samples_per_second": 8.195, "eval_steps_per_second": 0.546, "step": 70 }, { "epoch": 23.67, "learning_rate": 7.605263157894738e-05, "loss": 0.3447, "step": 71 }, { "epoch": 24.0, "learning_rate": 7.578947368421052e-05, "loss": 0.3733, "step": 72 }, { "epoch": 24.33, "learning_rate": 7.55263157894737e-05, "loss": 0.3481, "step": 73 }, { "epoch": 24.67, "learning_rate": 7.526315789473685e-05, "loss": 0.3377, "step": 74 }, { "epoch": 25.0, "learning_rate": 7.500000000000001e-05, "loss": 0.5311, "step": 75 }, { "epoch": 25.33, "learning_rate": 7.473684210526316e-05, "loss": 0.3228, "step": 76 }, { "epoch": 25.67, "learning_rate": 7.447368421052632e-05, "loss": 0.3248, "step": 77 }, { "epoch": 26.0, "learning_rate": 7.421052631578948e-05, "loss": 0.5215, "step": 78 }, { "epoch": 26.33, "learning_rate": 7.394736842105263e-05, "loss": 0.3395, "step": 79 }, { "epoch": 26.67, "learning_rate": 7.368421052631579e-05, "loss": 0.3136, "step": 80 }, { "epoch": 26.67, "eval_accuracy_dropoff": 0.33123693339766125, "eval_accuracy_undropoff": 0.9931389986088537, "eval_accuracy_unlabeled": NaN, "eval_iou_dropoff": 0.2885107958297481, "eval_iou_undropoff": 0.9632929658594979, "eval_iou_unlabeled": NaN, "eval_loss": 0.31975847482681274, "eval_mean_accuracy": 0.6621879660032575, "eval_mean_iou": 0.625901880844623, "eval_overall_accuracy": 0.9638313293457031, "eval_runtime": 1.7647, "eval_samples_per_second": 8.5, "eval_steps_per_second": 0.567, "step": 80 }, { "epoch": 27.0, "learning_rate": 7.342105263157895e-05, "loss": 0.4235, "step": 81 }, { "epoch": 27.33, "learning_rate": 7.315789473684212e-05, "loss": 0.3284, "step": 82 }, { "epoch": 27.67, "learning_rate": 7.289473684210527e-05, "loss": 0.3256, "step": 83 }, { "epoch": 28.0, "learning_rate": 7.263157894736842e-05, "loss": 0.2932, "step": 84 }, { "epoch": 28.33, "learning_rate": 7.236842105263159e-05, "loss": 0.3111, "step": 85 }, { "epoch": 28.67, "learning_rate": 7.210526315789474e-05, "loss": 0.2889, "step": 86 }, { "epoch": 29.0, "learning_rate": 7.18421052631579e-05, "loss": 0.4514, "step": 87 }, { "epoch": 29.33, "learning_rate": 7.157894736842105e-05, "loss": 0.2871, "step": 88 }, { "epoch": 29.67, "learning_rate": 7.131578947368421e-05, "loss": 0.2896, "step": 89 }, { "epoch": 30.0, "learning_rate": 7.105263157894737e-05, "loss": 0.2682, "step": 90 }, { "epoch": 30.0, "eval_accuracy_dropoff": 0.2984009924874216, "eval_accuracy_undropoff": 0.9956857967904649, "eval_accuracy_unlabeled": NaN, "eval_iou_dropoff": 0.2729809111973981, "eval_iou_undropoff": 0.9643402953348353, "eval_iou_unlabeled": NaN, "eval_loss": 0.29188063740730286, "eval_mean_accuracy": 0.6470433946389432, "eval_mean_iou": 0.6186606032661167, "eval_overall_accuracy": 0.9648114522298177, "eval_runtime": 1.7687, "eval_samples_per_second": 8.481, "eval_steps_per_second": 0.565, "step": 90 }, { "epoch": 30.33, "learning_rate": 7.078947368421054e-05, "loss": 0.2736, "step": 91 }, { "epoch": 30.67, "learning_rate": 7.052631578947368e-05, "loss": 0.2922, "step": 92 }, { "epoch": 31.0, "learning_rate": 7.026315789473685e-05, "loss": 0.4261, "step": 93 }, { "epoch": 31.33, "learning_rate": 7.000000000000001e-05, "loss": 0.2706, "step": 94 }, { "epoch": 31.67, "learning_rate": 6.973684210526316e-05, "loss": 0.2784, "step": 95 }, { "epoch": 32.0, "learning_rate": 6.947368421052632e-05, "loss": 0.282, "step": 96 }, { "epoch": 32.33, "learning_rate": 6.921052631578948e-05, "loss": 0.2683, "step": 97 }, { "epoch": 32.67, "learning_rate": 6.894736842105263e-05, "loss": 0.2584, "step": 98 }, { "epoch": 33.0, "learning_rate": 6.86842105263158e-05, "loss": 0.4286, "step": 99 }, { "epoch": 33.33, "learning_rate": 6.842105263157895e-05, "loss": 0.2521, "step": 100 }, { "epoch": 33.33, "eval_accuracy_dropoff": 0.3764215314632297, "eval_accuracy_undropoff": 0.9926302243822065, "eval_accuracy_unlabeled": NaN, "eval_iou_dropoff": 0.32476065885710886, "eval_iou_undropoff": 0.9647583860020452, "eval_iou_unlabeled": NaN, "eval_loss": 0.2957443594932556, "eval_mean_accuracy": 0.6845258779227181, "eval_mean_iou": 0.644759522429577, "eval_overall_accuracy": 0.9653457641601563, "eval_runtime": 1.8302, "eval_samples_per_second": 8.196, "eval_steps_per_second": 0.546, "step": 100 }, { "epoch": 33.67, "learning_rate": 6.81578947368421e-05, "loss": 0.2615, "step": 101 }, { "epoch": 34.0, "learning_rate": 6.789473684210527e-05, "loss": 0.3988, "step": 102 }, { "epoch": 34.33, "learning_rate": 6.763157894736843e-05, "loss": 0.2438, "step": 103 }, { "epoch": 34.67, "learning_rate": 6.736842105263159e-05, "loss": 0.2528, "step": 104 }, { "epoch": 35.0, "learning_rate": 6.710526315789474e-05, "loss": 0.2239, "step": 105 }, { "epoch": 35.33, "learning_rate": 6.68421052631579e-05, "loss": 0.2451, "step": 106 }, { "epoch": 35.67, "learning_rate": 6.657894736842106e-05, "loss": 0.2308, "step": 107 }, { "epoch": 36.0, "learning_rate": 6.631578947368421e-05, "loss": 0.3734, "step": 108 }, { "epoch": 36.33, "learning_rate": 6.605263157894737e-05, "loss": 0.2378, "step": 109 }, { "epoch": 36.67, "learning_rate": 6.578947368421054e-05, "loss": 0.2287, "step": 110 }, { "epoch": 36.67, "eval_accuracy_dropoff": 0.4591288165965952, "eval_accuracy_undropoff": 0.9920996303403997, "eval_accuracy_unlabeled": NaN, "eval_iou_dropoff": 0.3922413370101767, "eval_iou_undropoff": 0.9678471282288508, "eval_iou_unlabeled": NaN, "eval_loss": 0.2746734023094177, "eval_mean_accuracy": 0.7256142234684975, "eval_mean_iou": 0.6800442326195137, "eval_overall_accuracy": 0.9685007731119791, "eval_runtime": 1.7999, "eval_samples_per_second": 8.334, "eval_steps_per_second": 0.556, "step": 110 }, { "epoch": 37.0, "learning_rate": 6.55263157894737e-05, "loss": 0.2326, "step": 111 }, { "epoch": 37.33, "learning_rate": 6.526315789473684e-05, "loss": 0.2312, "step": 112 }, { "epoch": 37.67, "learning_rate": 6.500000000000001e-05, "loss": 0.2291, "step": 113 }, { "epoch": 38.0, "learning_rate": 6.473684210526316e-05, "loss": 0.3384, "step": 114 }, { "epoch": 38.33, "learning_rate": 6.447368421052632e-05, "loss": 0.2169, "step": 115 }, { "epoch": 38.67, "learning_rate": 6.421052631578948e-05, "loss": 0.2521, "step": 116 }, { "epoch": 39.0, "learning_rate": 6.394736842105263e-05, "loss": 0.2165, "step": 117 }, { "epoch": 39.33, "learning_rate": 6.368421052631579e-05, "loss": 0.2124, "step": 118 }, { "epoch": 39.67, "learning_rate": 6.342105263157896e-05, "loss": 0.2082, "step": 119 }, { "epoch": 40.0, "learning_rate": 6.31578947368421e-05, "loss": 0.2203, "step": 120 }, { "epoch": 40.0, "eval_accuracy_dropoff": 0.5471833574562915, "eval_accuracy_undropoff": 0.9902029029933593, "eval_accuracy_unlabeled": NaN, "eval_iou_dropoff": 0.4516702540227378, "eval_iou_undropoff": 0.9698565844064895, "eval_iou_unlabeled": NaN, "eval_loss": 0.2536683678627014, "eval_mean_accuracy": 0.7686931302248254, "eval_mean_iou": 0.7107634192146136, "eval_overall_accuracy": 0.9705869038899739, "eval_runtime": 1.8218, "eval_samples_per_second": 8.234, "eval_steps_per_second": 0.549, "step": 120 }, { "epoch": 40.33, "learning_rate": 6.289473684210526e-05, "loss": 0.2111, "step": 121 }, { "epoch": 40.67, "learning_rate": 6.263157894736843e-05, "loss": 0.2005, "step": 122 }, { "epoch": 41.0, "learning_rate": 6.236842105263159e-05, "loss": 0.3247, "step": 123 }, { "epoch": 41.33, "learning_rate": 6.210526315789474e-05, "loss": 0.1957, "step": 124 }, { "epoch": 41.67, "learning_rate": 6.18421052631579e-05, "loss": 0.2034, "step": 125 }, { "epoch": 42.0, "learning_rate": 6.157894736842106e-05, "loss": 0.1876, "step": 126 }, { "epoch": 42.33, "learning_rate": 6.131578947368421e-05, "loss": 0.1933, "step": 127 }, { "epoch": 42.67, "learning_rate": 6.105263157894737e-05, "loss": 0.1982, "step": 128 }, { "epoch": 43.0, "learning_rate": 6.078947368421053e-05, "loss": 0.3006, "step": 129 }, { "epoch": 43.33, "learning_rate": 6.052631578947369e-05, "loss": 0.1964, "step": 130 }, { "epoch": 43.33, "eval_accuracy_dropoff": 0.4166838973510695, "eval_accuracy_undropoff": 0.9941381864859773, "eval_accuracy_unlabeled": NaN, "eval_iou_dropoff": 0.3698843155549437, "eval_iou_undropoff": 0.9679789056516562, "eval_iou_unlabeled": NaN, "eval_loss": 0.23557429015636444, "eval_mean_accuracy": 0.7054110419185233, "eval_mean_iou": 0.6689316106033, "eval_overall_accuracy": 0.9685696919759115, "eval_runtime": 1.7453, "eval_samples_per_second": 8.595, "eval_steps_per_second": 0.573, "step": 130 }, { "epoch": 43.67, "learning_rate": 6.026315789473685e-05, "loss": 0.1787, "step": 131 }, { "epoch": 44.0, "learning_rate": 6e-05, "loss": 0.2787, "step": 132 }, { "epoch": 44.33, "learning_rate": 5.973684210526316e-05, "loss": 0.1813, "step": 133 }, { "epoch": 44.67, "learning_rate": 5.947368421052632e-05, "loss": 0.2007, "step": 134 }, { "epoch": 45.0, "learning_rate": 5.9210526315789485e-05, "loss": 0.1878, "step": 135 }, { "epoch": 45.33, "learning_rate": 5.8947368421052634e-05, "loss": 0.1832, "step": 136 }, { "epoch": 45.67, "learning_rate": 5.868421052631579e-05, "loss": 0.1739, "step": 137 }, { "epoch": 46.0, "learning_rate": 5.8421052631578954e-05, "loss": 0.1672, "step": 138 }, { "epoch": 46.33, "learning_rate": 5.815789473684211e-05, "loss": 0.1747, "step": 139 }, { "epoch": 46.67, "learning_rate": 5.789473684210526e-05, "loss": 0.1776, "step": 140 }, { "epoch": 46.67, "eval_accuracy_dropoff": 0.43434534886392356, "eval_accuracy_undropoff": 0.9930974877409892, "eval_accuracy_unlabeled": NaN, "eval_iou_dropoff": 0.3780242741741982, "eval_iou_undropoff": 0.9677365989228159, "eval_iou_unlabeled": NaN, "eval_loss": 0.22047756612300873, "eval_mean_accuracy": 0.7137214183024564, "eval_mean_iou": 0.672880436548507, "eval_overall_accuracy": 0.9683570861816406, "eval_runtime": 1.7567, "eval_samples_per_second": 8.539, "eval_steps_per_second": 0.569, "step": 140 }, { "epoch": 47.0, "learning_rate": 5.7631578947368423e-05, "loss": 0.1934, "step": 141 }, { "epoch": 47.33, "learning_rate": 5.736842105263159e-05, "loss": 0.1712, "step": 142 }, { "epoch": 47.67, "learning_rate": 5.710526315789474e-05, "loss": 0.1672, "step": 143 }, { "epoch": 48.0, "learning_rate": 5.684210526315789e-05, "loss": 0.2477, "step": 144 }, { "epoch": 48.33, "learning_rate": 5.6578947368421056e-05, "loss": 0.1671, "step": 145 }, { "epoch": 48.67, "learning_rate": 5.631578947368421e-05, "loss": 0.165, "step": 146 }, { "epoch": 49.0, "learning_rate": 5.6052631578947376e-05, "loss": 0.1969, "step": 147 }, { "epoch": 49.33, "learning_rate": 5.5789473684210526e-05, "loss": 0.1585, "step": 148 }, { "epoch": 49.67, "learning_rate": 5.552631578947369e-05, "loss": 0.1661, "step": 149 }, { "epoch": 50.0, "learning_rate": 5.5263157894736845e-05, "loss": 0.1675, "step": 150 }, { "epoch": 50.0, "eval_accuracy_dropoff": 0.4561651388793163, "eval_accuracy_undropoff": 0.9926318209540475, "eval_accuracy_unlabeled": NaN, "eval_iou_dropoff": 0.39357178961139355, "eval_iou_undropoff": 0.9682366151033369, "eval_iou_unlabeled": NaN, "eval_loss": 0.20607712864875793, "eval_mean_accuracy": 0.7243984799166818, "eval_mean_iou": 0.6809042023573653, "eval_overall_accuracy": 0.968878173828125, "eval_runtime": 1.8666, "eval_samples_per_second": 8.036, "eval_steps_per_second": 0.536, "step": 150 }, { "epoch": 50.33, "learning_rate": 5.500000000000001e-05, "loss": 0.1637, "step": 151 }, { "epoch": 50.67, "learning_rate": 5.473684210526316e-05, "loss": 0.153, "step": 152 }, { "epoch": 51.0, "learning_rate": 5.447368421052632e-05, "loss": 0.1516, "step": 153 }, { "epoch": 51.33, "learning_rate": 5.421052631578948e-05, "loss": 0.1604, "step": 154 }, { "epoch": 51.67, "learning_rate": 5.394736842105264e-05, "loss": 0.1433, "step": 155 }, { "epoch": 52.0, "learning_rate": 5.368421052631579e-05, "loss": 0.2353, "step": 156 }, { "epoch": 52.33, "learning_rate": 5.342105263157895e-05, "loss": 0.147, "step": 157 }, { "epoch": 52.67, "learning_rate": 5.315789473684211e-05, "loss": 0.1557, "step": 158 }, { "epoch": 53.0, "learning_rate": 5.2894736842105274e-05, "loss": 0.1458, "step": 159 }, { "epoch": 53.33, "learning_rate": 5.2631578947368424e-05, "loss": 0.148, "step": 160 }, { "epoch": 53.33, "eval_accuracy_dropoff": 0.4920279366829784, "eval_accuracy_undropoff": 0.991531250765024, "eval_accuracy_unlabeled": NaN, "eval_iou_dropoff": 0.41598764652752823, "eval_iou_undropoff": 0.9687330852459408, "eval_iou_unlabeled": NaN, "eval_loss": 0.19536948204040527, "eval_mean_accuracy": 0.7417795937240013, "eval_mean_iou": 0.6923603658867346, "eval_overall_accuracy": 0.9694142659505208, "eval_runtime": 1.8886, "eval_samples_per_second": 7.942, "eval_steps_per_second": 0.529, "step": 160 }, { "epoch": 53.67, "learning_rate": 5.236842105263158e-05, "loss": 0.1429, "step": 161 }, { "epoch": 54.0, "learning_rate": 5.2105263157894744e-05, "loss": 0.1908, "step": 162 }, { "epoch": 54.33, "learning_rate": 5.18421052631579e-05, "loss": 0.1435, "step": 163 }, { "epoch": 54.67, "learning_rate": 5.157894736842105e-05, "loss": 0.1393, "step": 164 }, { "epoch": 55.0, "learning_rate": 5.131578947368421e-05, "loss": 0.1601, "step": 165 }, { "epoch": 55.33, "learning_rate": 5.1052631578947376e-05, "loss": 0.136, "step": 166 }, { "epoch": 55.67, "learning_rate": 5.078947368421053e-05, "loss": 0.1401, "step": 167 }, { "epoch": 56.0, "learning_rate": 5.052631578947368e-05, "loss": 0.1505, "step": 168 }, { "epoch": 56.33, "learning_rate": 5.0263157894736846e-05, "loss": 0.1326, "step": 169 }, { "epoch": 56.67, "learning_rate": 5e-05, "loss": 0.1364, "step": 170 }, { "epoch": 56.67, "eval_accuracy_dropoff": 0.4927918303581685, "eval_accuracy_undropoff": 0.990152078789756, "eval_accuracy_unlabeled": NaN, "eval_iou_dropoff": 0.40640497922952673, "eval_iou_undropoff": 0.9674190748086308, "eval_iou_unlabeled": NaN, "eval_loss": 0.19154708087444305, "eval_mean_accuracy": 0.7414719545739622, "eval_mean_iou": 0.6869120270190787, "eval_overall_accuracy": 0.9681299845377604, "eval_runtime": 1.8388, "eval_samples_per_second": 8.157, "eval_steps_per_second": 0.544, "step": 170 }, { "epoch": 57.0, "learning_rate": 4.9736842105263166e-05, "loss": 0.2022, "step": 171 }, { "epoch": 57.33, "learning_rate": 4.9473684210526315e-05, "loss": 0.1299, "step": 172 }, { "epoch": 57.67, "learning_rate": 4.921052631578948e-05, "loss": 0.1349, "step": 173 }, { "epoch": 58.0, "learning_rate": 4.8947368421052635e-05, "loss": 0.2092, "step": 174 }, { "epoch": 58.33, "learning_rate": 4.86842105263158e-05, "loss": 0.1248, "step": 175 }, { "epoch": 58.67, "learning_rate": 4.842105263157895e-05, "loss": 0.1354, "step": 176 }, { "epoch": 59.0, "learning_rate": 4.815789473684211e-05, "loss": 0.1409, "step": 177 }, { "epoch": 59.33, "learning_rate": 4.789473684210527e-05, "loss": 0.1309, "step": 178 }, { "epoch": 59.67, "learning_rate": 4.763157894736843e-05, "loss": 0.1239, "step": 179 }, { "epoch": 60.0, "learning_rate": 4.736842105263158e-05, "loss": 0.1171, "step": 180 }, { "epoch": 60.0, "eval_accuracy_dropoff": 0.5733797413099915, "eval_accuracy_undropoff": 0.9898777345284205, "eval_accuracy_unlabeled": NaN, "eval_iou_dropoff": 0.47056771687689725, "eval_iou_undropoff": 0.9706919811185363, "eval_iou_unlabeled": NaN, "eval_loss": 0.17764781415462494, "eval_mean_accuracy": 0.781628737919206, "eval_mean_iou": 0.7206298489977168, "eval_overall_accuracy": 0.9714360555013021, "eval_runtime": 1.7716, "eval_samples_per_second": 8.467, "eval_steps_per_second": 0.564, "step": 180 }, { "epoch": 60.33, "learning_rate": 4.710526315789474e-05, "loss": 0.1263, "step": 181 }, { "epoch": 60.67, "learning_rate": 4.68421052631579e-05, "loss": 0.1235, "step": 182 }, { "epoch": 61.0, "learning_rate": 4.657894736842106e-05, "loss": 0.1369, "step": 183 }, { "epoch": 61.33, "learning_rate": 4.6315789473684214e-05, "loss": 0.1249, "step": 184 }, { "epoch": 61.67, "learning_rate": 4.605263157894737e-05, "loss": 0.1252, "step": 185 }, { "epoch": 62.0, "learning_rate": 4.578947368421053e-05, "loss": 0.1093, "step": 186 }, { "epoch": 62.33, "learning_rate": 4.552631578947369e-05, "loss": 0.1202, "step": 187 }, { "epoch": 62.67, "learning_rate": 4.526315789473684e-05, "loss": 0.1204, "step": 188 }, { "epoch": 63.0, "learning_rate": 4.5e-05, "loss": 0.1757, "step": 189 }, { "epoch": 63.33, "learning_rate": 4.473684210526316e-05, "loss": 0.1169, "step": 190 }, { "epoch": 63.33, "eval_accuracy_dropoff": 0.3741298504376594, "eval_accuracy_undropoff": 0.9964596019427087, "eval_accuracy_unlabeled": NaN, "eval_iou_dropoff": 0.347569272142274, "eval_iou_undropoff": 0.9683803113559507, "eval_iou_unlabeled": NaN, "eval_loss": 0.17537614703178406, "eval_mean_accuracy": 0.6852947261901841, "eval_mean_iou": 0.6579747917491123, "eval_overall_accuracy": 0.9689041137695312, "eval_runtime": 1.776, "eval_samples_per_second": 8.446, "eval_steps_per_second": 0.563, "step": 190 }, { "epoch": 63.67, "learning_rate": 4.4473684210526316e-05, "loss": 0.1242, "step": 191 }, { "epoch": 64.0, "learning_rate": 4.421052631578947e-05, "loss": 0.1529, "step": 192 }, { "epoch": 64.33, "learning_rate": 4.3947368421052636e-05, "loss": 0.1156, "step": 193 }, { "epoch": 64.67, "learning_rate": 4.368421052631579e-05, "loss": 0.1171, "step": 194 }, { "epoch": 65.0, "learning_rate": 4.342105263157895e-05, "loss": 0.11, "step": 195 }, { "epoch": 65.33, "learning_rate": 4.3157894736842105e-05, "loss": 0.1167, "step": 196 }, { "epoch": 65.67, "learning_rate": 4.289473684210527e-05, "loss": 0.1112, "step": 197 }, { "epoch": 66.0, "learning_rate": 4.263157894736842e-05, "loss": 0.109, "step": 198 }, { "epoch": 66.33, "learning_rate": 4.236842105263158e-05, "loss": 0.1028, "step": 199 }, { "epoch": 66.67, "learning_rate": 4.210526315789474e-05, "loss": 0.1178, "step": 200 }, { "epoch": 66.67, "eval_accuracy_dropoff": 0.454510993176649, "eval_accuracy_undropoff": 0.9921765318840718, "eval_accuracy_unlabeled": NaN, "eval_iou_dropoff": 0.38884766767071727, "eval_iou_undropoff": 0.9677201762977982, "eval_iou_unlabeled": NaN, "eval_loss": 0.16762010753154755, "eval_mean_accuracy": 0.7233437625303605, "eval_mean_iou": 0.6782839219842577, "eval_overall_accuracy": 0.9683698018391927, "eval_runtime": 1.8742, "eval_samples_per_second": 8.003, "eval_steps_per_second": 0.534, "step": 200 }, { "epoch": 67.0, "learning_rate": 4.1842105263157894e-05, "loss": 0.1549, "step": 201 }, { "epoch": 67.33, "learning_rate": 4.157894736842105e-05, "loss": 0.1064, "step": 202 }, { "epoch": 67.67, "learning_rate": 4.1315789473684214e-05, "loss": 0.1118, "step": 203 }, { "epoch": 68.0, "learning_rate": 4.105263157894737e-05, "loss": 0.0964, "step": 204 }, { "epoch": 68.33, "learning_rate": 4.078947368421053e-05, "loss": 0.1136, "step": 205 }, { "epoch": 68.67, "learning_rate": 4.0526315789473684e-05, "loss": 0.1065, "step": 206 }, { "epoch": 69.0, "learning_rate": 4.026315789473685e-05, "loss": 0.1578, "step": 207 }, { "epoch": 69.33, "learning_rate": 4e-05, "loss": 0.1081, "step": 208 }, { "epoch": 69.67, "learning_rate": 3.973684210526316e-05, "loss": 0.1091, "step": 209 }, { "epoch": 70.0, "learning_rate": 3.9473684210526316e-05, "loss": 0.1016, "step": 210 }, { "epoch": 70.0, "eval_accuracy_dropoff": 0.4025489925793186, "eval_accuracy_undropoff": 0.9943747452137437, "eval_accuracy_unlabeled": NaN, "eval_iou_dropoff": 0.35896398426616405, "eval_iou_undropoff": 0.967592274925927, "eval_iou_unlabeled": NaN, "eval_loss": 0.1669619232416153, "eval_mean_accuracy": 0.6984618688965312, "eval_mean_iou": 0.6632781295960455, "eval_overall_accuracy": 0.9681699117024739, "eval_runtime": 1.8214, "eval_samples_per_second": 8.235, "eval_steps_per_second": 0.549, "step": 210 }, { "epoch": 70.33, "learning_rate": 3.921052631578948e-05, "loss": 0.101, "step": 211 }, { "epoch": 70.67, "learning_rate": 3.894736842105263e-05, "loss": 0.1069, "step": 212 }, { "epoch": 71.0, "learning_rate": 3.868421052631579e-05, "loss": 0.1396, "step": 213 }, { "epoch": 71.33, "learning_rate": 3.842105263157895e-05, "loss": 0.1014, "step": 214 }, { "epoch": 71.67, "learning_rate": 3.8157894736842105e-05, "loss": 0.1028, "step": 215 }, { "epoch": 72.0, "learning_rate": 3.789473684210526e-05, "loss": 0.1477, "step": 216 }, { "epoch": 72.33, "learning_rate": 3.7631578947368425e-05, "loss": 0.099, "step": 217 }, { "epoch": 72.67, "learning_rate": 3.736842105263158e-05, "loss": 0.1028, "step": 218 }, { "epoch": 73.0, "learning_rate": 3.710526315789474e-05, "loss": 0.1256, "step": 219 }, { "epoch": 73.33, "learning_rate": 3.6842105263157895e-05, "loss": 0.1025, "step": 220 }, { "epoch": 73.33, "eval_accuracy_dropoff": 0.43655087646748, "eval_accuracy_undropoff": 0.9943074231011173, "eval_accuracy_unlabeled": NaN, "eval_iou_dropoff": 0.3887806200479793, "eval_iou_undropoff": 0.9690121216663343, "eval_iou_unlabeled": NaN, "eval_loss": 0.16480425000190735, "eval_mean_accuracy": 0.7154291497842986, "eval_mean_iou": 0.6788963708571568, "eval_overall_accuracy": 0.9696111043294271, "eval_runtime": 1.8392, "eval_samples_per_second": 8.156, "eval_steps_per_second": 0.544, "step": 220 }, { "epoch": 73.67, "learning_rate": 3.657894736842106e-05, "loss": 0.0986, "step": 221 }, { "epoch": 74.0, "learning_rate": 3.631578947368421e-05, "loss": 0.0958, "step": 222 }, { "epoch": 74.33, "learning_rate": 3.605263157894737e-05, "loss": 0.1012, "step": 223 }, { "epoch": 74.67, "learning_rate": 3.578947368421053e-05, "loss": 0.0966, "step": 224 }, { "epoch": 75.0, "learning_rate": 3.5526315789473684e-05, "loss": 0.0908, "step": 225 }, { "epoch": 75.33, "learning_rate": 3.526315789473684e-05, "loss": 0.1018, "step": 226 }, { "epoch": 75.67, "learning_rate": 3.5000000000000004e-05, "loss": 0.093, "step": 227 }, { "epoch": 76.0, "learning_rate": 3.473684210526316e-05, "loss": 0.0997, "step": 228 }, { "epoch": 76.33, "learning_rate": 3.447368421052632e-05, "loss": 0.0961, "step": 229 }, { "epoch": 76.67, "learning_rate": 3.421052631578947e-05, "loss": 0.0956, "step": 230 }, { "epoch": 76.67, "eval_accuracy_dropoff": 0.4279240471431525, "eval_accuracy_undropoff": 0.9927057954493445, "eval_accuracy_unlabeled": NaN, "eval_iou_dropoff": 0.3697151647479158, "eval_iou_undropoff": 0.9670745569523455, "eval_iou_unlabeled": NaN, "eval_loss": 0.1607496440410614, "eval_mean_accuracy": 0.7103149212962485, "eval_mean_iou": 0.6683948608501307, "eval_overall_accuracy": 0.9676984151204427, "eval_runtime": 1.9642, "eval_samples_per_second": 7.637, "eval_steps_per_second": 0.509, "step": 230 }, { "epoch": 77.0, "learning_rate": 3.3947368421052636e-05, "loss": 0.0909, "step": 231 }, { "epoch": 77.33, "learning_rate": 3.368421052631579e-05, "loss": 0.095, "step": 232 }, { "epoch": 77.67, "learning_rate": 3.342105263157895e-05, "loss": 0.0915, "step": 233 }, { "epoch": 78.0, "learning_rate": 3.3157894736842106e-05, "loss": 0.1312, "step": 234 }, { "epoch": 78.33, "learning_rate": 3.289473684210527e-05, "loss": 0.0894, "step": 235 }, { "epoch": 78.67, "learning_rate": 3.263157894736842e-05, "loss": 0.1006, "step": 236 }, { "epoch": 79.0, "learning_rate": 3.236842105263158e-05, "loss": 0.0902, "step": 237 }, { "epoch": 79.33, "learning_rate": 3.210526315789474e-05, "loss": 0.0919, "step": 238 }, { "epoch": 79.67, "learning_rate": 3.1842105263157895e-05, "loss": 0.0909, "step": 239 }, { "epoch": 80.0, "learning_rate": 3.157894736842105e-05, "loss": 0.1443, "step": 240 }, { "epoch": 80.0, "eval_accuracy_dropoff": 0.43322535437774257, "eval_accuracy_undropoff": 0.9936637385539103, "eval_accuracy_unlabeled": NaN, "eval_iou_dropoff": 0.3811034761519806, "eval_iou_undropoff": 0.968239431726135, "eval_iou_unlabeled": NaN, "eval_loss": 0.16110308468341827, "eval_mean_accuracy": 0.7134445464658264, "eval_mean_iou": 0.6746714539390578, "eval_overall_accuracy": 0.9688486735026042, "eval_runtime": 1.7665, "eval_samples_per_second": 8.491, "eval_steps_per_second": 0.566, "step": 240 } ], "max_steps": 360, "num_train_epochs": 120, "total_flos": 4.76793375031296e+16, "trial_name": null, "trial_params": null }