Closed redLinmumu closed 1 year ago
MutualInformation.py", line 14, in init self.bins = nn.Parameter(torch.linspace(0, 255, num_bins, device=device).float(), requires_grad=False) NameError: name 'device' is not defined
You can specify a global variable device=....
device=...
Nice catch! Fixed.
MutualInformation.py", line 14, in init self.bins = nn.Parameter(torch.linspace(0, 255, num_bins, device=device).float(), requires_grad=False) NameError: name 'device' is not defined