GaitGraph2 icon indicating copy to clipboard operation
GaitGraph2 copied to clipboard
verified publisher icon tteepe

GPU memory requirement

Open brianw0924 opened this issue 3 years ago • 0 comments

Hi, I'm wondering how much memory is required?

I've lower the batch_size to 1 and even decrease the embedding_layer

but still OOM with my RTX3080 (10G)

brianw0924 avatar Jul 20 '22 18:07 brianw0924