Update VBPR.py

duxy-me · Nov 27, 2018 · c924225 · c924225
1 parent 332bc5e
commit c924225
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/model/VBPR.py b/src/model/VBPR.py
@@ -92,7 +92,7 @@ def _create_loss(self):
             else:
                 print 'gradient noise'
                 self.delta = tf.gradients(self.loss, [self.image_feature])[0]
-            tf.stop_gradient(self.delta)
+            self.delta = tf.stop_gradient(self.delta)
 
             self.pos_pred_adv,_,_,_ = self._create_inference(self.user_input, self.pos_input, adv=True)
             self.neg_pred_adv,_,_,_ = self._create_inference(self.user_input, self.neg_input, adv=True)