fix None error

SeanLee97 · Feb 28, 2024 · 477c7e0 · 477c7e0
1 parent d31be12
commit 477c7e0
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/angle_emb/train_cli.py b/angle_emb/train_cli.py
@@ -171,6 +171,7 @@ def main():
             'fixed_teacher_name_or_path': args.fixed_teacher_name_or_path
         }
     if args.apply_tdmse:
+        trainer_kwargs = trainer_kwargs or {}
         trainer_kwargs = dict(trainer_kwargs, **{
             'apply_tdmse_kl': args.apply_tdmse_kl,
             'tdmse_kl_temperature': args.tdmse_kl_temperature,