fix TimeDistributedCriterion() lack of parameter of dimension issue (i…

…ntel-analytics#2940)
dding3 · Oct 24, 2019 · a912191 · a912191
1 parent 2d989e4
commit a912191
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...a/dllib/src/main/scala/com/intel/analytics/bigdl/dllib/utils/python/api/PythonBigDL.scala b/...a/dllib/src/main/scala/com/intel/analytics/bigdl/dllib/utils/python/api/PythonBigDL.scala
@@ -387,8 +387,8 @@ class PythonBigDL[T: ClassTag](implicit ev: TensorNumeric[T]) extends Serializab
   }
 
   def createTimeDistributedCriterion(critrn: TensorCriterion[T],
-    sizeAverage: Boolean = false): TimeDistributedCriterion[T] = {
-    TimeDistributedCriterion[T](critrn, sizeAverage)
+    sizeAverage: Boolean = false, dimension: Int = 2): TimeDistributedCriterion[T] = {
+    TimeDistributedCriterion[T](critrn, sizeAverage, dimension)
   }
 
   def createGRU(