deepinsight / deepinsight/insightface
训练数据中各ID图片数量不均，造成各显卡占用差异大
Open
- Dominant language
- Python
- Stars
- 29.7k
- Forks
- 6.1k
- PR merge metrics
- No merged PRs in 30d
Description
您好，我用多卡训练partialFC分类层，之前用50万分类时，四张显卡占用显存基本一致。但是当分类增加到116万时，有两张显卡的占用就特别的高(25G)，另外两张卡占用7G左右。这两张占用较多显存的卡分到的分类每个ID图片数量较多，平均50张，剩下两张占用很少显存的显卡分到的分类每个ID只有两三张图片，这种会造成各个显卡占用不均吗？
Contributor guide
No contributing guide indexed for this repository
Assessment
This issue has not been assessed yet.