Update modules/components/upr_net/upr.py
Browse files
modules/components/upr_net/upr.py
CHANGED
|
@@ -31,7 +31,7 @@ gaussian_kernel = torch.tensor([[1, 2, 1],
|
|
| 31 |
[2, 4, 2],
|
| 32 |
[1, 2, 1]]) / 16
|
| 33 |
gaussian_kernel = gaussian_kernel.repeat(2, 1, 1, 1)
|
| 34 |
-
gaussian_kernel = gaussian_kernel.to(torch.cuda.current_device())
|
| 35 |
|
| 36 |
|
| 37 |
def gaussian(x):
|
|
|
|
| 31 |
[2, 4, 2],
|
| 32 |
[1, 2, 1]]) / 16
|
| 33 |
gaussian_kernel = gaussian_kernel.repeat(2, 1, 1, 1)
|
| 34 |
+
gaussian_kernel = gaussian_kernel.to("cpu")#torch.cuda.current_device())
|
| 35 |
|
| 36 |
|
| 37 |
def gaussian(x):
|