Fix AUC changing output and target (intel-analytics#2541)

dding3 · Jul 8, 2020 · f05594e · f05594e
1 parent c710643
commit f05594e
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/spark/dl/src/main/scala/com/intel/analytics/bigdl/dllib/zooKeras/metrics/AUC.scala b/spark/dl/src/main/scala/com/intel/analytics/bigdl/dllib/zooKeras/metrics/AUC.scala
@@ -131,15 +131,15 @@ class AUC[T](thresholdNum: Int = 200)(implicit ev: TensorNumeric[T])
   override def apply(output: Activity, target: Activity):
   ValidationResult = {
     val _output = if (output.asInstanceOf[Tensor[T]].dim() == 2) {
-      output.asInstanceOf[Tensor[T]].squeeze(2)
+      output.asInstanceOf[Tensor[T]].clone().squeeze(2)
     } else {
-      output.asInstanceOf[Tensor[T]].squeeze()
+      output.asInstanceOf[Tensor[T]].clone().squeeze()
     }
 
     val _target = if (target.asInstanceOf[Tensor[T]].dim() == 2) {
-      target.asInstanceOf[Tensor[T]].squeeze(2)
+      target.asInstanceOf[Tensor[T]].clone().squeeze(2)
     } else {
-      target.asInstanceOf[Tensor[T]].squeeze()
+      target.asInstanceOf[Tensor[T]].clone().squeeze()
     }
     require(_output.dim() <= 2 && _target.dim() <= 2,
       s"${_output.dim()} dim format is not supported")