I am trying to run my cnn project on GPU. I tried to change the tensor type:
def _evaluate_epoch(axes, tr_loader, val_loader, model, criterion, epoch, stats):
    torch.set_default_tensor_type('torch.cuda.FloatTensor')

and in main function I added
    model = model.cuda()

after the model was created. But the project still doesn’t seem to run on GPU. Is there anything wrong? I am pretty new to this. Thanks!