Tweak repr

This commit is contained in:
Jensun Ravichandran 2021-06-02 01:07:48 +02:00
parent fcf3a4979c
commit 7743c50725

View File

@ -126,7 +126,7 @@ class ConnectionTopology(torch.nn.Module):
self.age[b][a] = 0
def extra_repr(self):
return f"agelimit: {self.agelimit}"
return f"(agelimit): ({self.agelimit})"
class KNN(AbstractPrototypeModel):