From cf37f1404b39432f42545ace119d5d57f25be0a8 Mon Sep 17 00:00:00 2001
From: KaiyangZhou <k.zhou@qmul.ac.uk>
Date: Thu, 3 May 2018 15:01:29 +0100
Subject: [PATCH] add loss

---
 models/Inception.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/models/Inception.py b/models/Inception.py
index aac1174..61f7631 100644
--- a/models/Inception.py
+++ b/models/Inception.py
@@ -343,6 +343,7 @@ def inceptionv4(num_classes=1000, pretrained='imagenet'):
 class InceptionV4ReID(nn.Module):
     def __init__(self, num_classes, loss={'xent'}, **kwargs):
         super(InceptionV4ReID, self).__init__()
+        self.loss = loss
         base = inceptionv4()
         self.features = base.features
         self.classifier = nn.Linear(1536, num_classes)