search

mit-han-lab / bevfusion

[ICRA'23] BEVFusion: Multi-Task Multi-Sensor Fusion with Unified Bird's-Eye View Representation
https://bevfusion.mit.edu
Apache License 2.0
2.26k stars 409 forks source link

multi-gpu training in Docker #549

Closed huiyegit closed 10 months ago

huiyegit commented 10 months ago

Thank you for providing the Docker environment. The original command for multi-gpu training is as follows:

   torchpack dist-run -np 8 python tools/train.py configs/nuscenes/det/centerhead/lssfpn/camera/256x704/swint/default.yaml --model.encoders.camera.backbone.init_cfg.checkpoint pretrained/swint-nuimages-pretrained.pth

what is the command for multi-gpu training in Docker?

huiyegit commented 10 months ago

The same command also works in Docker. Thank you.

huiyegit commented 10 months ago

Thank you