Upload train_config.toml with huggingface_hub
Browse files- train_config.toml +3 -3
train_config.toml
CHANGED
|
@@ -37,16 +37,16 @@ max_grad_norm = 1.0
|
|
| 37 |
lr_warmup_steps = 250
|
| 38 |
|
| 39 |
# === Duration ===
|
| 40 |
-
max_train_steps =
|
| 41 |
|
| 42 |
# === Save ===
|
| 43 |
-
save_every_n_steps =
|
| 44 |
output_dir = "/workspace/projects/paste_pose_first/outputs/20260325_033320"
|
| 45 |
output_name = "paste_pose_first"
|
| 46 |
|
| 47 |
# === Sampling ===
|
| 48 |
sample_prompts = "/workspace/projects/paste_pose_first/sample_prompts.txt"
|
| 49 |
-
sample_every_n_steps =
|
| 50 |
|
| 51 |
# === Dataloader ===
|
| 52 |
max_data_loader_n_workers = 2
|
|
|
|
| 37 |
lr_warmup_steps = 250
|
| 38 |
|
| 39 |
# === Duration ===
|
| 40 |
+
max_train_steps = 5000
|
| 41 |
|
| 42 |
# === Save ===
|
| 43 |
+
save_every_n_steps = 250
|
| 44 |
output_dir = "/workspace/projects/paste_pose_first/outputs/20260325_033320"
|
| 45 |
output_name = "paste_pose_first"
|
| 46 |
|
| 47 |
# === Sampling ===
|
| 48 |
sample_prompts = "/workspace/projects/paste_pose_first/sample_prompts.txt"
|
| 49 |
+
sample_every_n_steps = 150
|
| 50 |
|
| 51 |
# === Dataloader ===
|
| 52 |
max_data_loader_n_workers = 2
|