the GPU you used for trainning

[howstrangy]

Hello, I started to train this project on Rat7m , using GTX 4060 8G , and CUDA was out of memory.
may I ask what GPU were you using to train this project?

[jenjsun]

Apologies for the late response! You could try reducing the batch size to see if it helps.

We used either 1 Amazon p2 or p3 for training on the cloud. The p2 is the smaller of the two I believe
Amazon p2: https://aws.amazon.com/ec2/instance-types/p2/
Amazon p3: https://aws.amazon.com/ec2/instance-types/p3/