INNER CODE UNIT · Python
validate_save
adobe/antialiased-cnns · main.py:657
def validate_save(val_loader, mean, std, args):
import matplotlib.pyplot as plt
import os
for i, (input, target) in enumerate(val_loader):
img = (255*np.clip(input[0,...].data.cpu().numpy()*np.array(std)[:,None,None] + mean[:,None,None],0,1)).astype('uint8').transpose((1,2,0))
plt.imsave(os.path.join(args.out_dir,'%05d.png'%i),img)
# def save_checkpoint(state, is_best, filename='checkpoint.pth.tar'):
def save_checkpoint(state, is_best, epoch, out_dir='./'):
torch.save(state, os.path.join(out_dir,'checkpoint.pth.tar'))
if(epoch % 10 == 0):
torch.save(state, os.path.join(out_dir,'checkpoint_%03d.pth.tar'%epoch))
if is_best:
shutil.copyfile(os.path.join(out_dir,'checkpoint.pth.tar'), os.path.join(out_dir,'model_best.pth.tar'))
class AverageMeter(object):
"""Computes and stores the average and current value"""