search

megvii-model / SinglePathOneShot

MIT License
259 stars 53 forks source link

代码运行时间 #12

Closed nbl97 closed 3 years ago

nbl97 commented 3 years ago

请问默认 Supernet/train.py 在 imagenet上要运行多久？我用两块V100，20个小时只跑了 8880 Ite。预计完整跑完需要两周？查看了一下，发现GPU的利用率很低，不到10%。是哪里设置的有问题吗？不知你们如何解决的？原版代码，没有改动。

nbl97 commented 3 years ago

目前发现是因为默认dataloader的num_workers=1，将其设为32速度大幅提升。还有别的需要注意的地方吗？作者训练超网络需要多久？