Commit 5cc4b714 authored by brianhhu's avatar brianhhu

Clean up resnet model definition

parent 9d4b9b7b
import torch
import numpy as np
from foolbox.models import PyTorchModel
......@@ -8,8 +7,7 @@ from resnet18.resnet_model import Model
def create_model():
# instantiate the model
resnet18 = Model()
resnet18.load_state_dict(torch.load('./resnet18/converted_pytorch.pt'))
resnet18 = Model(pretrained=True)
resnet18.cuda().eval()
# preprocessing
......
torch==0.3.1
torchvision==0.2.0
foolbox==1.1.0
git+https://github.com/bveliqi/adversarial-vision-challenge
This diff is collapsed.
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment