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
|