SAEHD: write_preview_history now works faster

The frequency at which the preview is saved now depends on the resolution. For example 64x64 – every 10 iters. 448x448 – every 70 iters.
2020-07-18 11:03:13 +04:00
parent 58670722dc
commit ce95af9068
2 changed files with 65 additions and 19 deletions
@@ -609,6 +609,10 @@ Examples: df, liae, df-d, df-ud, liae-ud, ...
        for model, filename in io.progress_bar_generator(self.get_model_filename_list(), "Saving", leave=False):
            model.save_weights ( self.get_strpath_storage_for_file(filename) )

+    #override
+    def should_save_preview_history(self):
+        return (not io.is_colab() and self.iter % ( 10*(max(1,self.resolution // 64)) ) == 0) or \
+               (io.is_colab() and self.iter % 100 == 0)

    #override
    def onTrainOneIter(self):