AlibabaResearch / AlibabaResearch/AdvancedLiterateMachinery

VGT training memory requirement

Abierto
#198 0 comentarios 0 reacciones 0 asignados Ver en GitHub
Lenguaje dominante
C++
Estrellas
1.8k
Forks
195
Métricas de merge de PR
Sin PR fusionados en 30 d

Descripción

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.

Guía de contribución

No hay ninguna guía de contribución indexada para este repositorio

Evaluación

Este issue todavía no se ha evaluado.

Recibe los nuevos issues en tu correo

Un resumen breve de issues de GitHub para principiantes.