# GPU memory for training

yuval-alaluf / SAM

Official Implementation for "Only a Matter of Style: Age Transformation Using a Style-Based Regression Model" (SIGGRAPH 2021) https://arxiv.org/abs/2102.02754

https://yuval-alaluf.github.io/SAM/

MIT License

632 stars 151 forks source link

# GPU memory for training #11

Closed vinduon closed 3 years ago

vinduon commented 3 years ago

Hi sir!

I do not know how much GPU used for training with batch_size = 2?

yuval-alaluf commented 3 years ago

Hi @duongquangvinh I dont have a precise estimate but probably not too much (less than 10GB maybe?). The best way to find out is to run the script :smile:

vinduon commented 3 years ago

Thank you for your response, sir.