CUDA memory is related to the size of dataset

tianrun-chen / SAM-Adapter-PyTorch

Adapting Meta AI's Segment Anything to Downstream Tasks with Adapters and Prompts

MIT License

968 stars 83 forks source link

CUDA memory is related to the size of dataset #24

Closed TengliEd closed 1 year ago

TengliEd commented 1 year ago

Hi @tianrun-chen , I can train the adapted model on CAMO-1k on 4*32G v100 while out of memory error was thrown on COD10k with batchsize = 1. It's because of the data preparation mechanism?