From 5eaa59cea6edaeb75ad46a735f89613386026645 Mon Sep 17 00:00:00 2001 From: Colombo Date: Mon, 9 Mar 2020 16:42:05 +0400 Subject: [PATCH] _ --- facelib/TernausNet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/facelib/TernausNet.py b/facelib/TernausNet.py index 9ac54c0..08088f6 100644 --- 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):