您好，我这边在用tusimple数据集跑Bezierlannet的训练时，命令是 python main_landet.py --train --config=configs/lane_detection/bezierlanenet/resnet18_tusimple_aug1b.py，出现了报错，请问您是否能提供解决方案或者建议呢？谢谢您！ #180
Description
Traceback (most recent call last):
File "main_landet.py", line 76, in
runner.run()
File "/home/ubuntu/code/pytorch-auto-drive/utils/runners/lane_det_trainer.py", line 55, in run
self.model)
File "/home/ubuntu/anaconda3/envs/pad/lib/python3.6/site-packages/torch/nn/modules/module.py", line 722, in _call_impl
result = self.forward(*input, **kwargs)
File "/home/ubuntu/code/pytorch-auto-drive/utils/losses/hungarian_bezier_loss.py", line 150, in forward
loss_seg = self.binary_seg_loss(inputs=outputs['segmentations'], targets=target_segmentations)
File "/home/ubuntu/code/pytorch-auto-drive/utils/losses/hungarian_bezier_loss.py", line 204, in binary_seg_loss
reduction='none') / self.pos_weight_seg
File "/home/ubuntu/anaconda3/envs/pad/lib/python3.6/site-packages/torch/nn/functional.py", line 2538, in binary_cross_entropy_with_logits
raise ValueError("Target size ({}) must be the same as input size ({})".format(target.size(), input.size()))
ValueError: Target size (torch.Size([20, 360, 640, 3])) must be the same as input size (torch.Size([20, 640, 3]))