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

Minor aesthetic change

Browse Source
This commit is contained in:
Jensun Ravichandran 2021-06-05 01:23:58 +02:00
parent b82bb54dbe
commit bed753a6e9
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/callbacks.py
View File

@ -44,7 +44,7 @@ class PruneLoserPrototypes(pl.Callback):
if self.verbose: if self.verbose:
print(f"\nPrototype win ratios: {ratios}") print(f"\nPrototype win ratios: {ratios}")
print(f"Pruning prototypes at: {to_prune}") print(f"Pruning prototypes at: {to_prune}")
print(f"Corresponding labels are: {prune_labels}") print(f"Corresponding labels are: {prune_labels.tolist()}")
cur_num_protos = pl_module.num_prototypes cur_num_protos = pl_module.num_prototypes
pl_module.remove_prototypes(indices=to_prune) pl_module.remove_prototypes(indices=to_prune)
if self.replace: if self.replace: