revanb88

torch.cuda.OutOfMemoryError: CUDA out of memory.

2

023-09-23 02:12:42.767428: Epoch 0 2023-09-23 02:12:42.767778: Current learning rate: 0.01 using pin_memory on device 0 Traceback (most recent call last): File "/data/revan/miniconda3/envs/nnUNet/bin/nnUNetv2_train", line 8, in sys.exit(run_training_entry()) File "/data/revan/experiments/CMR_experiments/nnUNetFrame/nnUNet/nnunetv2/run/run_training.py", line 268,...