Aquila-Consortium/map2map
32
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix stupid merging bug

Browse Source
This commit is contained in:
Yin Li 2020-02-09 23:07:40 -05:00
parent 16b82fcc56
commit b41e85eda5
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
map2map/train.py
View File

@ -240,7 +240,7 @@ def gpu_worker(local_rank, node, args):
logger.close() logger.close()
good = min_loss is None or epoch_loss[0] < min_loss[0] good = min_loss is None or epoch_loss[0] < min_loss[0]
if good and epoch >= args.adv_delay: if good:
min_loss = epoch_loss min_loss = epoch_loss
state = { state = {