INNER CODE UNIT · Python

global_step

adobe/antialiased-cnns · main.py:478

                global_step = i + (epoch * len(train_loader))
                wandb.log(
                    {
                        'train_loss': losses.val,
                        'train_avg_loss': losses.avg,
                        'train_acc@1': top1.val,
                        'train_avg_acc@1': top1.avg,
                        'train_acc@5': top5.val,
                        'train_avg_acc@5': top5.avg,
                        'epoch': 1.*global_step/len(train_loader), 
                    },
                    step=global_step)

        if(i > args.max_train_iters):
            break

def validate(val_loader, model, criterion, args):
    batch_time = AverageMeter()

View source record →