top1.update(prec1[0], inputs.size(0))
top5.update(prec5[0], inputs.size(0))
if i % args.print_freq == 0:
print_logger.info(
'Epoch[{0}]({1}/{2}):\t'
'Loss_sparse {loss_sparse.val:.4f} ({loss_sparse.avg:.4f})\t'
'Loss_data {loss_data.val:.4f} ({loss_data.avg:.4f})\t'
'Loss_d {loss_d.val:.4f} ({loss_d.avg:.4f})\t'
'Loss_g {loss_g.val:.4f} ({loss_g.avg:.4f})\t'
'Prec@1 {top1.val:.4f} \t'
'Prec@5 {top5.val:.4f} '.format(
epoch, i, num_iterations, loss_sparse=losses_sparse, loss_data=losses_data, loss_d=losses_d, loss_g=losses_g, top1=top1, top5=top5))
改为
top1.update(prec1.item(), inputs.size(0))
top5.update(prec5.item(), inputs.size(0))