Cuda memory error

[narges-tk]

Hi!
May I please ask you to suggest the configuration needed for running the script? While running the model for S3DIS dataset, I am facing te following error.

[zhang-zihui]

Hi, @narges-tk ,

We use a single 3090GPU to do all experiments, so it requires 24G cuda memory.

[narges-tk]

Thanks for your prompt reply!

[xwmxcz]

Hello, if the training is interrupted, how can I continue the previous epoch training? I wish you academic success.

[zihui0930]

Hello, if the training is interrupted, how can I continue the previous epoch training? I wish you academic success.

Hi, we didn't implement the resume code, you may add it on your own. You can save the model weight and optimizer, and reload them for subsequent training.