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

Fix ImageCBC bug

Browse Source
This commit is contained in:
Jensun Ravichandran 2021-05-21 13:11:36 +02:00
parent 88a34a06ef
commit 0933a88a1b
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/cbc.py
View File

@ -153,4 +153,4 @@ class ImageCBC(CBC):
"""
def on_train_batch_end(self, outputs, batch, batch_idx, dataloader_idx):
# super().on_train_batch_end(outputs, batch, batch_idx, dataloader_idx)
self.component_layer.prototypes.data.clamp_(0.0, 1.0)
self.component_layer.components.data.clamp_(0.0, 1.0)