Migrations
/
pytorch-image-models
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

log loss as before

Browse Source
pull/550/head
Aman Arora 4 years ago
parent 624c9b6949
commit a9e5d9e5ad
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File

4
train.py
Unescape View File

@ -723,7 +723,7 @@ def train_one_epoch(
if hasattr(optimizer, 'sync_lookahead'): if hasattr(optimizer, 'sync_lookahead'):
optimizer.sync_lookahead() optimizer.sync_lookahead()
return OrderedDict([('train_loss', losses_m.avg)]) return OrderedDict([('loss', losses_m.avg)])
def validate(model, loader, loss_fn, args, amp_autocast=suppress, log_suffix=''): def validate(model, loader, loss_fn, args, amp_autocast=suppress, log_suffix=''):
@ -785,7 +785,7 @@ def validate(model, loader, loss_fn, args, amp_autocast=suppress, log_suffix='')
log_name, batch_idx, last_idx, batch_time=batch_time_m, log_name, batch_idx, last_idx, batch_time=batch_time_m,
loss=losses_m, top1=top1_m, top5=top5_m)) loss=losses_m, top1=top1_m, top5=top5_m))
metrics = OrderedDict([('val_loss', losses_m.avg), ('top1', top1_m.avg), ('top5', top5_m.avg)]) metrics = OrderedDict([('loss', losses_m.avg), ('top1', top1_m.avg), ('top5', top5_m.avg)])
return metrics return metrics

Write Preview
Loading…
Cancel
Save