From ca7119e677e14b209b224fafe4de57780113499f Mon Sep 17 00:00:00 2001 From: Jordan Gong Date: Sun, 3 Jan 2021 19:00:07 +0800 Subject: Update hyperparameter configuration, implement prototype fit function --- models/rgb_part_net.py | 1 - 1 file changed, 1 deletion(-) (limited to 'models/rgb_part_net.py') 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: -- cgit v1.2.3