Spaces:
Sleeping
Sleeping
Christian J. Steinmetz
commited on
Commit
·
0b430bb
1
Parent(s):
9ff30f9
in the default config log the model to wandb
Browse files- cfg/config.yaml +2 -0
cfg/config.yaml
CHANGED
|
@@ -63,6 +63,7 @@ datamodule:
|
|
| 63 |
shuffle_removed_effects: ${shuffle_removed_effects}
|
| 64 |
render_files: ${render_files}
|
| 65 |
render_root: ${render_root}
|
|
|
|
| 66 |
val_dataset:
|
| 67 |
_target_: remfx.datasets.EffectDataset
|
| 68 |
total_chunks: 1000
|
|
@@ -109,6 +110,7 @@ logger:
|
|
| 109 |
job_type: "train"
|
| 110 |
group: ""
|
| 111 |
save_dir: "."
|
|
|
|
| 112 |
|
| 113 |
trainer:
|
| 114 |
_target_: pytorch_lightning.Trainer
|
|
|
|
| 63 |
shuffle_removed_effects: ${shuffle_removed_effects}
|
| 64 |
render_files: ${render_files}
|
| 65 |
render_root: ${render_root}
|
| 66 |
+
parallel: True
|
| 67 |
val_dataset:
|
| 68 |
_target_: remfx.datasets.EffectDataset
|
| 69 |
total_chunks: 1000
|
|
|
|
| 110 |
job_type: "train"
|
| 111 |
group: ""
|
| 112 |
save_dir: "."
|
| 113 |
+
log_model: True
|
| 114 |
|
| 115 |
trainer:
|
| 116 |
_target_: pytorch_lightning.Trainer
|