How to set the config to use multi node and multi GPUS?

PKU-YuanGroup / Video-LLaVA

Video-LLaVA: Learning United Visual Representation by Alignment Before Projection

https://arxiv.org/pdf/2311.10122.pdf

Apache License 2.0

2.65k stars 192 forks source link

Open rovebot opened 3 months ago

rovebot commented 3 months ago

Hi, How to set the training config to use multi node and multi GPUS?

rhelck commented 1 month ago

@rovebot Did you try to set the device_map ?