From 895a06731a2353e26290088d0f1ad89150a1fafa Mon Sep 17 00:00:00 2001
From: nieyan <951946218@qq.com>
Date: Mon, 9 Mar 2020 14:13:56 +0800
Subject: [PATCH] Fix parameter passing bug.

---
 code/lib/model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/code/lib/model.py b/code/lib/model.py
index a2ac57b..9c442c0 100644
--- a/code/lib/model.py
+++ b/code/lib/model.py
@@ -112,7 +112,7 @@ def __define_criterion(self, class_weights, delta_var,
         # Discriminative Loss
         if self.use_instance_segmentation:
             self.criterion_discriminative = DiscriminativeLoss(
-                delta_var, delta_dist, norm, self.usegpu)
+                delta_var, delta_dist, norm, usegpu=self.usegpu)
             if self.usegpu:
                 self.criterion_discriminative = \
                     self.criterion_discriminative.cuda()