Memory usuage during validation

TXH-mercury / VAST

Code and Model for VAST: A Vision-Audio-Subtitle-Text Omni-Modality Foundation Model and Dataset

https://arxiv.org/abs/2305.18500

MIT License

241 stars 17 forks source link

Memory usuage during validation #9

Open 1980x opened 9 months ago

1980x commented 9 months ago

Hi, When validation set size is large, GPU memory usage is much more than required during training even with small batch size. Can you please suggest where is the issue? Thanks