I used GTX 3090 Ti to train the model , and i'm encountering the "CUDA out of memory" error when training reached iteration 1199. Then , i set the --N_rand to 8 , the --chunk to 64 , the --netchunk to 64 , the events_N_rand to128 , the --kernel_ptnum to 3. The error still occur. What should i do to solve this problem?
Dear Author,
I used GTX 3090 Ti to train the model , and i'm encountering the "CUDA out of memory" error when training reached iteration 1199. Then , i set the --N_rand to 8 , the --chunk to 64 , the --netchunk to 64 , the events_N_rand to128 , the --kernel_ptnum to 3. The error still occur. What should i do to solve this problem?