VGT training memory requirement

未关闭

#198 创建于 2024年11月25日 0 条评论 0 个 reaction 已指派 0 人在 GitHub 查看

描述

I am able to train it with one 2080Ti with 12 GB of VRAM but I would have to resort to having a batch size of 1.

Training with a batch size of 4 and 4 2080Tis throws out of memory after some time even with mixed precision enabled.

Please let me know about the:
1) Normal memory requirements for training
2) Any ways to reduce the memory required.

Thanks very much in advance.

这个仓库没有索引到贡献指南

这个 Issue 还没有评估数据。