I tried to place it here
# train val parameters dict
    net.load_state_dict(torch.load("/content/drive/My Drive/WS_DAN_PyTorch-master/checkpoint/car/checkpoint.pth.tar"))
    state = {'model': net, 'train_loader': train_loader,
             'val_loader': val_loader, 'criterion': criterion,
             'center': center_dict['center'], 'config': config,
             'optimizer': optimizer}

but it gave a runtime error, missing weights. I tried to add strict= False, it removed the error but it started from the beginning, it ignored the checkpoint!
image717×66 2.63 KB
yesterday it was at 14%. ( the epoch takes 6hours to run)