github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-06 04:52:13 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

_

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

2
facelib/TernausNet.py
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):