OOM when allocating tensor with shape[32,1024,32,64]

[overfitover]

when I run train_and_eval_scannet.py，I meet the error. Some idea.

[jmydurant]

Your GPU memory runs out. You can use small batch size or multiple GPU cards for training.