sh train.sh TypeError : cannot creat 'generator' instances

[wangziniu1109]

someone can help me solve the problem? There is a problemwhen i run train,sh

/home/lab201/wangziniu/D4LCN-master/lib/lr.py:100: UserWarning: Detected call of lr_scheduler.step() before optimizer.step(). In PyTorch 1.1.0 and later, you should call them in the opposite order: optimizer.step() before lr_scheduler.step(). Failure to do this will result in PyTorch skipping the first value of the learning rate schedule.See more details at https://pytorch.org/docs/stable/optim.html#how-to-adjust-learning-rate
"https://pytorch.org/docs/stable/optim.html#how-to-adjust-learning-rate", UserWarning)
Traceback (most recent call last):
File "/home/lab201/wangziniu/D4LCN-master/scripts/train.py", line 215, in
main(sys.argv[1:])
File "/home/lab201/wangziniu/D4LCN-master/scripts/train.py", line 128, in main
cls, prob, bbox_2d, bbox_3d, feat_size = rpn_net(images.cuda(), depths.cuda())
File "/home/lab201/anaconda3/envs/D4LCN/lib/python3.6/site-packages/torch/nn/modules/module.py", line 493, in call
result = self.forward(*input, **kwargs)
File "/home/lab201/anaconda3/envs/D4LCN/lib/python3.6/site-packages/torch/nn/parallel/data_parallel.py", line 153, in forward
return self.gather(outputs, self.output_device)
File "/home/lab201/anaconda3/envs/D4LCN/lib/python3.6/site-packages/torch/nn/parallel/data_parallel.py", line 165, in gather
return gather(outputs, output_device, dim=self.dim)
File "/home/lab201/anaconda3/envs/D4LCN/lib/python3.6/site-packages/torch/nn/parallel/scatter_gather.py", line 67, in gather
return gather_map(outputs)
File "/home/lab201/anaconda3/envs/D4LCN/lib/python3.6/site-packages/torch/nn/parallel/scatter_gather.py", line 62, in gather_map
return type(out)(map(gather_map, zip(*outputs)))
TypeError: cannot create 'generator' instances