feat(poly-norm): add default value for eps argument

torch-ext/activation/layers.py

@@ -5,7 +5,7 @@ from .poly_norm import PolyNormFunction
 
 
 class PolyNorm(nn.Module):
-    def __init__(self, eps):
+    def __init__(self, eps=1e-6):
         super().__init__()
         self.weight = torch.nn.Parameter(torch.ones(3) / 3)
         self.bias = torch.nn.Parameter(torch.zeros(1))