VGT training memory requirement

オープン

#198 2024年11月25日に作成コメント 0 件リアクション 0 件担当者 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 はまだ評価されていません。