[fix]

data_loader.py

@@ -15,6 +15,7 @@ class MyDataset(torch.utils.data.Dataset):
         for line in data_txt:
             line = line.strip()
             words = line.split(' ')
+            print(words)
             imgs.append((words[0], int(words[1].strip())))
         self.imgs = imgs
         self.transform = transform

utils.py

@@ -16,6 +16,11 @@ def load_model(model_name, pretrain=True, require_grad=True, num_class=1000, pre
         #for param in net.parameters():
             #param.requires_grad = require_grad
         net = PRENet(net, 512, num_class)
+    elif model_name == 'resnet152':
+        net = resnet152(pretrained=pretrain, path=pretrained_model)
+        #for param in net.parameters():
+            #param.requires_grad = require_grad
+        net = PRENet(net, 512, num_class)
 
     return net