Hi,
I started training the following model, using the train_mtl.py:

  
      
      GitHub
  
  
    

anilbatra2185/road_connectivity 5

Improved Road Connectivity by Joint Learning of Orientation and Segmentation (CVPR2019) - anilbatra2185/road_connectivity


  
  
    
    
  
  


After I entered training epoch 45 (out of 120), I received the following error:
Traceback (most recent call last):
  File "train_mtl.py", line 472, in <module>
    main()
  File "train_mtl.py", line 466, in main
    val_loss = test(epoch)
  File "train_mtl.py", line 367, in test
    loss2 = angle_loss(pred_vecmaps, util.to_variable(labels[0], True))
  File "C:\Users\...\road_connectivity-master\venv\lib\site-packages\torch\nn\modules\module.py", line 477, in __call__
    result = self.forward(*input, **kwargs)
  File "C:\Users\...\road_connectivity-master\utils\loss.py", line 15, in forward
    log_p = F.log_softmax(inputs, dim=1)
  File "C:\Users\...\road_connectivity-master\venv\lib\site-packages\torch\nn\functional.py", line 975, in log_softmax
    return input.log_softmax(dim)
RuntimeError: CUDA error: out of memory

I am running PyTorch 0.4.1 on Windows 10 with NVIDIA Quadro P3200 with Max-Q Design.
Any ideas on how to solve this?