Christian J. Steinmetz commited on
Commit
0b430bb
·
1 Parent(s): 9ff30f9

in the default config log the model to wandb

Browse files
Files changed (1) hide show
  1. 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