I am using google colab to run my deep learning model. I dont know why is it taking more time in colab/
I am using this snippet for device mounting in training the model.  Plz suggest something , its taking 30 min for 1 epoch.
‘’’
if args.use_gpu:
use_cuda = torch.cuda.is_available()

device = torch.device("cuda" if use_cuda else "cpu")

else:
device='cpu'

‘’’
@ptrblck plz help sir