AlibabaResearch / AlibabaResearch/AdvancedLiterateMachinery

VGT training memory requirement

Offen
#198 0 Kommentare 0 Reaktionen 0 zugewiesene Personen Auf GitHub ansehen
Vorherrschende Sprache
C++
Sterne
1.8k
Forks
195
PR-Merge-Kennzahlen
Keine gemergten PRs in 30 T.

Beschreibung

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.

Beitragsleitfaden

Für dieses Repository ist kein Beitragsleitfaden indexiert

Bewertung

Dieses Issue wurde noch nicht bewertet.

Neue Issues direkt in Ihr Postfach

Eine kurze Übersicht über anfängerfreundliche GitHub-Issues.