opendilab / InterFuser

[CoRL 2022] InterFuser: Safety-Enhanced Autonomous Driving Using Interpretable Sensor Fusion Transformer
Apache License 2.0
514 stars 42 forks source link

Minor issues in the code #64

Closed Aintky2000 closed 10 months ago

Aintky2000 commented 11 months ago

In train.py, line 973 & 982, maybe you can add "if args.distributed" in case that you run the code with single GPU, or there will be a bug.

deepcs233 commented 11 months ago

Hi! Sorry, I have never ran the code without the distributed mode. If we only use one gpu, the value of torch.distributed.get_world_size() maybe 1?