search

tteepe / GaitGraph2

Official code for "Towards a Deeper Understanding of Skeleton-based Gait Recognition" (CVPRW'22)
40 stars 11 forks source link

GPU memory requirement #10

Open brianw0924 opened 2 years ago

brianw0924 commented 2 years ago

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)