fix DFLJPG,

SAE: added "rare sample booster"
SAE: pixel loss replaced to smooth transition from DSSIM to PixelLoss in 15k epochs by default
This commit is contained in:
iperov committed 2019-02-09 18:53:37 +04:00
1 parent f93b4713a9
commit 4d37fd62cd
11 files changed
+174 -101

No files matched your search

+1 -1
View File
@@ -55,7 +55,7 @@ class Model(ModelBase):
[self.decoder_dst, self.get_strpath_storage_for_file(self.decoder_dstH5)]] )
#override
def onTrainOneEpoch(self, sample):
def onTrainOneEpoch(self, sample, generators_list):
warped_src, target_src, target_src_mask = sample[0]
warped_dst, target_dst, target_dst_mask = sample[1]