finetune fuyu8b text location with image size of 1920x1080 always got OOM even on A100*8

#63
by Nooodles - opened

I used huggingface Trainer+deepspeed(stage3) to finetune it and I had to downsample the image to 1536x864 to avoid OOM.
Is there any way to use the original resolution to finetune?

Sign up or log in to comment