Update modules/components/upr_net_multi_flow/upr.py
Browse files
modules/components/upr_net_multi_flow/upr.py
CHANGED
@@ -22,7 +22,7 @@ gaussian_kernel = torch.tensor([[1, 2, 1],
|
|
22 |
[2, 4, 2],
|
23 |
[1, 2, 1]]) / 16
|
24 |
gaussian_kernel = gaussian_kernel.repeat(2, 1, 1, 1)
|
25 |
-
gaussian_kernel = gaussian_kernel.to(torch.cuda.current_device())
|
26 |
|
27 |
|
28 |
def gaussian(x):
|
|
|
22 |
[2, 4, 2],
|
23 |
[1, 2, 1]]) / 16
|
24 |
gaussian_kernel = gaussian_kernel.repeat(2, 1, 1, 1)
|
25 |
+
gaussian_kernel = gaussian_kernel.to("cpu")#torch.cuda.current_device())
|
26 |
|
27 |
|
28 |
def gaussian(x):
|