summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorGard Spreemann <gspr@nonempty.org>2020-01-13 19:43:11 +0100
committerGard Spreemann <gspr@nonempty.org>2020-01-13 19:43:11 +0100
commit5814d359aae98dd0e8f89254efd9ece9fb40f5e6 (patch)
tree9eae850ad057c83121e32930b4133854e88520e2
parent7a3e06da1f629d1186f1f66dc3055f95a121ae40 (diff)
__repr__ for SCNN.HEADmaster
-rw-r--r--scnn/scnn.py4
1 files changed, 4 insertions, 0 deletions
diff --git a/scnn/scnn.py b/scnn/scnn.py
index f46cc56..f825cb9 100644
--- a/scnn/scnn.py
+++ b/scnn/scnn.py
@@ -24,6 +24,7 @@ class SimplicialConvolution(nn.Module):
self.C_out = C_out
self.K = K
self.enable_bias = enable_bias
+ self.variance = variance
self.theta = nn.parameter.Parameter(variance*torch.randn((self.C_out, self.C_in, self.K)))
if self.enable_bias:
@@ -46,6 +47,9 @@ class SimplicialConvolution(nn.Module):
return y + self.bias
+ def __repr__(self):
+ return "SimplicialConvolution(K=%d, C_in=%d, C_out=%d, enable_bias=%s, variance=%f)" %(self.K, self.C_in, self.C_out, self.enable_bias, self.variance)
+
# This class does not yet implement the
# Laplacian-power-pre/post-composed with the coboundary. It can be
# simulated by just adding more layers anyway, so keeping it simple