julius/prototorch_models
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Tweak repr

Browse Source
This commit is contained in:
Jensun Ravichandran 2021-06-02 01:07:48 +02:00
parent fcf3a4979c
commit 7743c50725
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
prototorch/models/unsupervised.py
View File

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