search

raoyongming / GFNet

[NeurIPS 2021] [T-PAMI] Global Filter Networks for Image Classification
https://gfnet.ivg-research.xyz/
MIT License
445 stars 41 forks source link

多卡训练 #9

Closed ZZM11 closed 3 years ago

ZZM11 commented 3 years ago

请问能多卡训练吗？我没有看到相关的代码---

raoyongming commented 3 years ago

Hi, thanks for your interest in our work.

Our code is deigned for training with multiple GPUs. The example command to train our model using PyTorch DDP with 8 GPUs: python -m torch.distributed.launch --nproc_per_node=8 --use_env main_gfnet.py --output_dir logs/gfnet-xs --arch gfnet-xs --batch-size 128 --data-path /path/to/ILSVRC2012/