fix update preview samples after disable pretrain

2020-01-28 13:32:01 +04:00
parent 7386a9d6fd
commit 9c6ca24642
3 changed files with 61 additions and 52 deletions
@@ -337,11 +337,12 @@ class QModel(ModelBase):

        # Loading/initializing all models/optimizers weights
        for model, filename in io.progress_bar_generator(self.model_filename_list, "Initializing models"):
-            do_init = self.is_first_run()
-
            if self.pretrain_just_disabled:
+                do_init = False
                if model == self.inter:
                    do_init = True
+            else:
+                do_init = self.is_first_run()

            if not do_init:
                do_init = not model.load_weights( self.get_strpath_storage_for_file(filename) )