CalvinBackup/iperov-DeepFaceLab
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

_

Browse Source
This commit is contained in:
Colombo
2020-03-09 16:42:05 +04:00
parent b0b9072981
commit 5eaa59cea6
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
facelib/TernausNet.py
+1 -1
View File
@@ -161,7 +161,7 @@ class TernausNet(object):
io.log_err("Unable to load VGG11 pretrained weights from vgg11_enc_weights.npy") io.log_err("Unable to load VGG11 pretrained weights from vgg11_enc_weights.npy")
def save_weights(self): 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 ) model.save_weights( self.weights_file_root / filename )
def extract (self, input_image): def extract (self, input_image):