Fixing small issue with default decay param.

AlgorithmicDataReduction · Jan 12, 2024 · 65a4695 · 65a4695
1 parent b75821b
commit 65a4695
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/torch_quadconv/quadconv.py b/torch_quadconv/quadconv.py
@@ -3,6 +3,7 @@
 
 import torch
 import torch.nn as nn
+import numpy as np
 
 from .utils.misc import Sin
 
@@ -56,7 +57,7 @@ def __init__(self,*,
 
         #decay parameter
         if decay_param == None:
-            self.decay_param = 4/torch.sqrt(self.in_points)
+            self.decay_param = 4/np.sqrt(self.in_points)
         else:
             self.decay_param = decay_param