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)