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

Add diff between out and tgt to figure

Browse Source
This commit is contained in:
Yin Li 2020-02-05 21:19:03 -05:00
parent e20f3194a5
commit 291dfb24b3
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
map2map/train.py
View File

@ -330,7 +330,8 @@ def train(epoch, loader, model, criterion, optimizer, scheduler,
args.logger.add_figure('fig/epoch/train/in',
fig3d(narrow_like(input, output)[-1]), global_step =epoch+1)
args.logger.add_figure('fig/epoch/train/out',
fig3d(output[-1, skip_chan:], target[-1, skip_chan:]),
fig3d(output[-1, skip_chan:], target[-1, skip_chan:],
output[-1, skip_chan:] - target[-1, skip_chan:]),
global_step =epoch+1)
return epoch_loss
@ -398,7 +399,8 @@ def validate(epoch, loader, model, criterion, adv_model, adv_criterion, args):
args.logger.add_figure('fig/epoch/val/in',
fig3d(narrow_like(input, output)[-1]), global_step =epoch+1)
args.logger.add_figure('fig/epoch/val',
fig3d(output[-1, skip_chan:], target[-1, skip_chan:]),
fig3d(output[-1, skip_chan:], target[-1, skip_chan:],
output[-1, skip_chan:] - target[-1, skip_chan:]),
global_step =epoch+1)
return epoch_loss