Upload policy weights, train config and readme
Browse files- model.safetensors +1 -1
- train_config.json +3 -3
model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 1051837384
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0bfbdd8f7f63f45d507ceb5ef0440060d6258e2497d4683fa5135ca4352e8312
|
| 3 |
size 1051837384
|
train_config.json
CHANGED
|
@@ -146,15 +146,15 @@
|
|
| 146 |
"scheduler_name": "cosine",
|
| 147 |
"scheduler_warmup_steps": 500
|
| 148 |
},
|
| 149 |
-
"output_dir": "outputs/train/
|
| 150 |
"job_name": "diffusion_92_ep",
|
| 151 |
"resume": false,
|
| 152 |
"seed": 1000,
|
| 153 |
"num_workers": 4,
|
| 154 |
"batch_size": 16,
|
| 155 |
-
"steps":
|
| 156 |
"eval_freq": 2000,
|
| 157 |
-
"log_freq":
|
| 158 |
"save_checkpoint": true,
|
| 159 |
"save_freq": 2000,
|
| 160 |
"use_policy_training_preset": true,
|
|
|
|
| 146 |
"scheduler_name": "cosine",
|
| 147 |
"scheduler_warmup_steps": 500
|
| 148 |
},
|
| 149 |
+
"output_dir": "outputs/train/diffusion_so100_test2",
|
| 150 |
"job_name": "diffusion_92_ep",
|
| 151 |
"resume": false,
|
| 152 |
"seed": 1000,
|
| 153 |
"num_workers": 4,
|
| 154 |
"batch_size": 16,
|
| 155 |
+
"steps": 100000,
|
| 156 |
"eval_freq": 2000,
|
| 157 |
+
"log_freq": 1000,
|
| 158 |
"save_checkpoint": true,
|
| 159 |
"save_freq": 2000,
|
| 160 |
"use_policy_training_preset": true,
|