updated hydra config

igcraft-model/.hydra/config.yaml

@@ -1,27 +1,3 @@
-trainer:
-  _target_: pytorch_lightning.Trainer
-  accelerator: auto
-  devices: auto
-  precision: 16-mixed
-  gradient_clip_val: 1.0
-  enable_progress_bar: false
-  log_every_n_steps: 5
-  val_check_interval: null
-  limit_val_batches: null
-  strategy:
-    _target_: pytorch_lightning.strategies.DDPStrategy
-    find_unused_parameters: true
-  logger:
-    _target_: pytorch_lightning.loggers.WandbLogger
-  callbacks:
-  - _target_: pytorch_lightning.callbacks.ModelCheckpoint
-    monitor: val_loss
-    mode: min
-    save_top_k: 3
-    dirpath: ${hydra:run.dir}/checkpoints
-    save_last: true
-  max_epochs: 250
-  accumulate_grad_batches: 4
 model:
   network:
     backbone_heavy:
@@ -131,12 +107,6 @@ model:
     transition_steps: 40000
     ema_decay: 0.999
     num_eval_time_bins: 10
-    out_dir: ${hydra:run.dir}
     vh_ckpt_path: null
     vl_ckpt_path: null
   _target_: igcraft.model.PairedSequenceBFN
-wandb:
-  entity: null
-  project: null
-  run_name: null
-  tags: null