first

network.py

@@ -11,7 +11,7 @@ import torch
 import os
 import matplotlib.pyplot as plt
 import numpy as np
-os.environ['CUDA_VISIBLE_DEVICE'] = '6'
+os.environ['CUDA_VISIBLE_DEVICE'] = '5'
 
 class MCNN(nn.Module):
     def __init__(self, load_vgg=True):