里面的resnet架构和一般的不一样，res20,res32,res44,res56,res110，如果不存在的话那代码块
if args.dataset!='imagenet': if args.arch=='resnet110': pretrain = torch.load(os.path.join(args.pretrain_path,'cifar_pretrained_nets/','resnet110.pth.tar')) elif args.arch=='resnet56': pretrain = torch.load(os.path.join(args.pretrain_path,'cifar_pretrained_nets/','resnet56.pth.tar')) elif args.arch=='resnet32': pretrain = torch.load(os.path.join(args.pretrain_path,'cifar_pretrained_nets/','resnet32.pth.tar')) elif args.arch=='resnet20': pretrain = torch.load(os.path.join(args.pretrain_path,'cifar_pretrained_nets/','resnet20.pth.tar')) net.load_state_dict(pretrain['state_dict'].state_dict())
是不是可以考虑注释掉了？
里面的resnet架构和一般的不一样，res20,res32,res44,res56,res110，如果不存在的话那代码块
if args.dataset!='imagenet': if args.arch=='resnet110': pretrain = torch.load(os.path.join(args.pretrain_path,'cifar_pretrained_nets/','resnet110.pth.tar')) elif args.arch=='resnet56': pretrain = torch.load(os.path.join(args.pretrain_path,'cifar_pretrained_nets/','resnet56.pth.tar')) elif args.arch=='resnet32': pretrain = torch.load(os.path.join(args.pretrain_path,'cifar_pretrained_nets/','resnet32.pth.tar')) elif args.arch=='resnet20': pretrain = torch.load(os.path.join(args.pretrain_path,'cifar_pretrained_nets/','resnet20.pth.tar')) net.load_state_dict(pretrain['state_dict'].state_dict())是不是可以考虑注释掉了？