SuyeonJ commited on
Commit
cbd456a
·
verified ·
1 Parent(s): 8224065

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):