AttributeError: 'RandomSampler' object has no attribute 'set_epoch'

uber-research / LaneGCN

[ECCV2020 Oral] Learning Lane Graph Representations for Motion Forecasting

https://arxiv.org/abs/2007.13732

Other

496 stars 131 forks source link

AttributeError: 'RandomSampler' object has no attribute 'set_epoch' #26

Closed SHITIANYU-hue closed 2 years ago

SHITIANYU-hue commented 2 years ago

When i run:

python train1.py -m lanegcn. it will have this issue:

SHITIANYU-hue commented 2 years ago

@chenyuntc

chenyuntc commented 2 years ago

Hi @SHITIANYU-hue. You may remove this line code. This is for shuffle in distributed training. Sorry for the inconvenience. Feel free to reopen it if you still have issues.