AlibabaResearch / AlibabaResearch/AdvancedLiterateMachinery

VGT training memory requirement

Đang mở
#198 0 bình luận 0 reaction 0 người được giao Xem trên GitHub
Ngôn ngữ chính
C++
Star
1.8k
Fork
195
Chỉ số merge pull request
Không có pull request nào được merge trong 30 ngày

Mô tả

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.

Hướng dẫn đóng góp

Chưa lập chỉ mục được hướng dẫn đóng góp cho kho mã nguồn này

Đánh giá

Issue này chưa được đánh giá.

Nhận issue mới trong hộp thư của bạn

Bản tóm tắt ngắn những issue GitHub phù hợp với người mới.