Skip to content

Commit a88a6d8

Browse files
committed
Fix the problem of duplicating name in dataloaderaw.
1 parent 0aeb222 commit a88a6d8

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

dataloaderraw.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020
])
2121

2222
from misc.resnet_utils import myResnet
23-
import misc.resnet as resnet
23+
import misc.resnet
2424

2525
class DataLoaderRaw():
2626

@@ -34,9 +34,9 @@ def __init__(self, opt):
3434

3535
# Load resnet
3636
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)
37+
self.my_resnet = getattr(misc.resnet, self.cnn_model)()
38+
self.my_resnet.load_state_dict(torch.load('./data/imagenet_weights/'+self.cnn_model+'.pth'))
39+
self.my_resnet = myResnet(self.my_resnet)
4040
self.my_resnet.cuda()
4141
self.my_resnet.eval()
4242

0 commit comments

Comments
 (0)