Memory usage on single A100*80GB in training

#36

by DavidWu1116 - opened Apr 11

Apr 11

Could it fit in a single A100*80GB without FSDP? I tried load it with deepspeed zero2 with a few lora params, but it always OOM.

Upload images, audio, and videos by dragging in the text input, pasting, or clicking here.

Tap or paste here to upload images

· Sign up or log in to comment