_

2021-06-01 00:21:27 +04:00
parent 315f241c51
commit 1981ed0ca8
1 changed files with 4 additions and 1 deletions
@@ -43,7 +43,10 @@ def trainerThread (s2c, c2s, e,
            if not saved_models_path.exists():
                saved_models_path.mkdir(exist_ok=True, parents=True)
-
+            
            if dump_ckpt:
                cpu_only=True
            model = models.import_model(model_class_name)(
                        is_training=not dump_ckpt,
                        saved_models_path=saved_models_path,