Error when running stage_1.py under Places_LT

[Zheweiqiu]

Hi, I am going to train on my own dataset, I changed all the data path accordingly. But when I was trying to fine-tune resnet-152, I encountered the following error and don't know how to fix it. Any help will be appreciate!

Traceback (most recent call last): File "main.py", line 57, in training_model.train() File "/data/OLTR/OpenLongTailRecognition-OLTR/run_networks.py", line 192, in train for step, (inputs, labels, _) in enumerate(self.data['train']): File "/opt/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 819, in next return self._process_data(data) File "/opt/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 846, in _process_data data.reraise() File "/opt/anaconda3/lib/python3.6/site-packages/torch/_utils.py", line 369, in reraise raise self.exc_type(msg) TypeError: function takes exactly 5 arguments (1 given)