CUDA not used

[marcomameli1992]

Dear Why do you not use CUDA?

[giacomopiccinini]

Hi, I do in fact use CUDA.

Quoting from Code.Modules.A_train (similarly, the batch are moved to the GPU, check the same file):

  # Enable CUDA and set up GPU
  if torch.cuda.is_available():
      model.cuda()

  model.train()
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
  model.to(device=device)