Update test.py

myavartanoo · Apr 22, 2022 · 0ee12fd · 0ee12fd
1 parent 108c84e
commit 0ee12fd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test.py b/test.py
@@ -57,7 +57,7 @@ def main(config):
                 if key!='directory':
                     target[key] = target[key].to('cuda') 
 
-            polycoeff, _, _, A_10x10 = model(img_H)
+            polycoeff, _, A_10x10 = model(img_H)
 
             polyorder = torch.from_numpy(gen_polynomial_orders(4)).to('cuda')
             PI_funcs_inout = PI_funcs_generator(target['inoutpts'], polycoeff, polyorder)