problems in train.py

[YUDASHUAI916]

Hi. When I run 'python train.py config/pan/pan_r18_ic15.py' , the errors are as followings:
Do you know how to solve the problem? Thank you very much.
Traceback (most recent call last):
File "train.py", line 234, in
main(args)
File "train.py", line 216, in main
train(train_loader, model, optimizer, epoch, start_iter, cfg)
File "train.py", line 41, in train
for iter, data in enumerate(train_loader):
File "D:\Anaconda3\lib\site-packages\torch\utils\data\dataloader.py", line 435, in next
data = self._next_data()
File "D:\Anaconda3\lib\site-packages\torch\utils\data\dataloader.py", line 1085, in _next_data
return self._process_data(data)
File "D:\Anaconda3\lib\site-packages\torch\utils\data\dataloader.py", line 1111, in _process_data
data.reraise()
File "D:\Anaconda3\lib\site-packages\torch_utils.py", line 428, in reraise
raise self.exc_type(msg)
TypeError: function takes exactly 5 arguments (1 given)

[whai362]

It may be the problem of software environment.

[qqqqq127]

same problem.