UNDIAL

Author: dong-river

src/trainer/__init__.py

@@ -21,7 +21,6 @@ def _register_trainer(trainer_class):
 
 def load_trainer_args(trainer_args: DictConfig, dataset):
     trainer_args = dict(trainer_args)
-    trainer_args["output_dir"] = trainer_args.pop("output_dir", "./output")
     warmup_epochs = trainer_args.pop("warmup_epochs", None)
     if warmup_epochs:
         batch_size = trainer_args["per_device_train_batch_size"]