From 1981ed0ca8d8e374264d6649a9c5546d7bb88a83 Mon Sep 17 00:00:00 2001 From: iperov Date: Tue, 1 Jun 2021 00:21:27 +0400 Subject: [PATCH] _ --- mainscripts/Trainer.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/mainscripts/Trainer.py b/mainscripts/Trainer.py index 66afd71..780cc48 100644 --- a/mainscripts/Trainer.py +++ b/mainscripts/Trainer.py @@ -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,