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

change 'epoch' to 'iter',

Browse source 
added timestamp prefix to training string
This commit is contained in:
iperov 2019-03-12 19:23:52 +04:00
parent 69174a48e0
commit 97b6fabaab
7 changed files with 93 additions and 87 deletions
Download patch file Download diff file Expand all files Collapse all files

4
models/Model_LIAEF128/Model.py
View file

@ -12,7 +12,7 @@ class Model(ModelBase):
def onInitializeOptions(self, is_first_run, ask_override):
if is_first_run or ask_override:
def_pixel_loss = self.options.get('pixel_loss', False)
self.options['pixel_loss'] = io.input_bool ("Use pixel loss? (y/n, ?:help skip: n/default ) : ", def_pixel_loss, help_message="Default DSSIM loss good for initial understanding structure of faces. Use pixel loss after 20k epochs to enhance fine details and decrease face jitter.")
self.options['pixel_loss'] = io.input_bool ("Use pixel loss? (y/n, ?:help skip: n/default ) : ", def_pixel_loss, help_message="Default DSSIM loss good for initial understanding structure of faces. Use pixel loss after 20k iters to enhance fine details and decrease face jitter.")
else:
self.options['pixel_loss'] = self.options.get('pixel_loss', False)
@ -70,7 +70,7 @@ class Model(ModelBase):
[self.inter_AB, 'inter_AB.h5']] )
#override
def onTrainOneEpoch(self, sample, generators_list):
def onTrainOneIter(self, sample, generators_list):
warped_src, target_src, target_src_mask = sample[0]
warped_dst, target_dst, target_dst_mask = sample[1]