_

2020-03-09 16:42:05 +04:00
parent b0b9072981
commit 5eaa59cea6
1 changed files with 1 additions and 1 deletions
--- a/facelib/TernausNet.py
+++ b/facelib/TernausNet.py
@@ -161,7 +161,7 @@ class TernausNet(object):
                        io.log_err("Unable to load VGG11 pretrained weights from vgg11_enc_weights.npy")

    def save_weights(self):
-        for model, filename in io.progress_bar_generator(self.model_filename_list, "Saving"):
+        for model, filename in io.progress_bar_generator(self.model_filename_list, "Saving", leave=False):
            model.save_weights( self.weights_file_root / filename )

    def extract (self, input_image):