We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 0aeb222 commit a88a6d8Copy full SHA for a88a6d8
dataloaderraw.py
@@ -20,7 +20,7 @@
20
])
21
22
from misc.resnet_utils import myResnet
23
-import misc.resnet as resnet
+import misc.resnet
24
25
class DataLoaderRaw():
26
@@ -34,9 +34,9 @@ def __init__(self, opt):
34
35
# Load resnet
36
self.cnn_model = opt.get('cnn_model', 'resnet101')
37
- resnet = getattr(resnet, self.cnn_model)()
38
- resnet.load_state_dict(torch.load('./data/imagenet_weights/'+self.cnn_model+'.pth'))
39
- self.my_resnet = myResnet(resnet)
+ self.my_resnet = getattr(misc.resnet, self.cnn_model)()
+ self.my_resnet.load_state_dict(torch.load('./data/imagenet_weights/'+self.cnn_model+'.pth'))
+ self.my_resnet = myResnet(self.my_resnet)
40
self.my_resnet.cuda()
41
self.my_resnet.eval()
42
0 commit comments