Atul Nagane

Issues while training on google colab

2

Traceback (most recent call last): File "train_PBAFN_stage1.py", line 30, in Traceback (most recent call last): File "train_PBAFN_stage1.py", line 30, in torch.cuda.set_device(opt.local_rank) File "/usr/local/lib/python3.7/dist-packages/torch/cuda/init.py", line 261, in set_device torch._C._cuda_setDevice(device) RuntimeError: CUDA...

Cuda Memory Error

1

I am getting this error RuntimeError: CUDA out of memory. Tried to allocate 648.00 MiB (GPU 0; 3.82 GiB total capacity; 1.57 GiB already allocated; 665.06 MiB free; 1.98 GiB...

nice code ... can we get multiple probability array? can we get prediction score and list of the array with a score like this good : 0.82 bad : 0.8...

training issue

1

could you provide me the custom training code? I'm facing issues while training the model.