lasso-net / lassonet

I think there is a bug in the following lines:

Lines 132 to 134 in 3b3b529

    
           self.device = device 
        
           if device is None: 
        
               device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

device is set to CUDA is torch reports that cuda is available, but its done after self.device is already set.

Thanks! Btw I didn't test it on GPU and I think that loading models might fail so please report me the bugs!

	self.device = device
	if device is None:
	device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

Check for presence of cuda is not being saved to self