I've been looking for a long time, but I can't find where to download the pth such as cifar_wrn_28.pth.
elif model_name == 'wrn28':
TRAINED_MODEL_PATH = data_path_join('pretrained_models/wrn_adv/')
filename = 'cifar_wrn_28.pth'
pretrained_model = wrn.WideNet()
pretrained_model = torch.nn.DataParallel(pretrained_model)
checkpoint = torch.load(os.path.join(TRAINED_MODEL_PATH, filename))
# if hasattr(pretrained_model, 'module'):
# pretrained_model = pretrained_model.module
pretrained_model.load_state_dict(checkpoint['net'])
Download links of pretrained weights can be found in Supplementary Sec.II of our paper.
bang bang wo!
I've been looking for a long time, but I can't find where to download the pth such as cifar_wrn_28.pth.
Download links of pretrained weights can be found in Supplementary Sec.II of our paper.
bang bang wo!