Update train_config.json
Browse files- train_config.json +2 -2
train_config.json
CHANGED
|
@@ -107,8 +107,6 @@
|
|
| 107 |
]
|
| 108 |
}
|
| 109 |
},
|
| 110 |
-
"device": "mps",
|
| 111 |
-
"use_amp": false,
|
| 112 |
"push_to_hub": true,
|
| 113 |
"repo_id": "raihanfsti/my_policy",
|
| 114 |
"private": null,
|
|
@@ -142,6 +140,8 @@
|
|
| 142 |
"optimizer_weight_decay": 0.0001,
|
| 143 |
"optimizer_lr_backbone": 1e-05
|
| 144 |
},
|
|
|
|
|
|
|
| 145 |
"output_dir": "outputs/train/record_test",
|
| 146 |
"job_name": "app-test",
|
| 147 |
"resume": false,
|
|
|
|
| 107 |
]
|
| 108 |
}
|
| 109 |
},
|
|
|
|
|
|
|
| 110 |
"push_to_hub": true,
|
| 111 |
"repo_id": "raihanfsti/my_policy",
|
| 112 |
"private": null,
|
|
|
|
| 140 |
"optimizer_weight_decay": 0.0001,
|
| 141 |
"optimizer_lr_backbone": 1e-05
|
| 142 |
},
|
| 143 |
+
"device": "mps",
|
| 144 |
+
"use_amp": false,
|
| 145 |
"output_dir": "outputs/train/record_test",
|
| 146 |
"job_name": "app-test",
|
| 147 |
"resume": false,
|