How much GPU is needed

[liangli66]

Excuse me.in paper,when training you set N = 3 ,D = 128, epochs=6 Batch size = 6 on 2 NVIDIA TITAN RTX graphics cards.But,in code train.sh
you set N = 4 ,D = 196, epochs=10 Batch size = 1 .When I attempt to train this network follow your train.sh ,batch =2 on 2 NVIDIA 3090(24G),11 days if training ends and GPU utilization below 46%，very slow.if batch=3,out cuda memory.why?