diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2021-01-03 19:00:07 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2021-01-03 19:00:07 +0800 |
| commit | ca7119e677e14b209b224fafe4de57780113499f (patch) | |
| tree | e829b91d3e4e39589aa22dd4f35ad4c2568880a3 /models/rgb_part_net.py | |
| parent | 425af1da453203a3d5b526d3e30af9c9f9faaa72 (diff) | |
Update hyperparameter configuration, implement prototype fit function
Diffstat (limited to 'models/rgb_part_net.py')
| -rw-r--r-- | models/rgb_part_net.py | 1 |
1 files changed, 0 insertions, 1 deletions
diff --git a/models/rgb_part_net.py b/models/rgb_part_net.py index 02345d6..0f3b4f4 100644 --- a/models/rgb_part_net.py +++ b/models/rgb_part_net.py @@ -76,7 +76,6 @@ class RGBPartNet(nn.Module): if self.training: # TODO Implement Batch All triplet loss function batch_all_triplet_loss = torch.tensor(0.) - print(*losses, batch_all_triplet_loss) loss = torch.sum(torch.stack((*losses, batch_all_triplet_loss))) return loss else: |