Add torchmodel and torchloss (intel-analytics#2224)

* add torchmodel and torchloss * add some comments * delete import tfnet * add ut * speed up set weight and get gradient * add unit test * fix style check * fix python style check * add test tag * some change * move torchmodel and loss to another folder * support input and target from scala * fix style * update model and los * fix ut * update * exclude torchmodel serial test
dding3 · May 9, 2020 · 94bf8d4 · 94bf8d4
1 parent ef5ea70
commit 94bf8d4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/bigdl/dllib/src/main/scala/com/intel/analytics/bigdl/dllib/feature/FeatureSet.scala b/bigdl/dllib/src/main/scala/com/intel/analytics/bigdl/dllib/feature/FeatureSet.scala
@@ -399,7 +399,7 @@ object PythonLoaderFeatureSet{
     jepRDD
   }
 
-  protected def toArrayTensor(
+  private[zoo] def toArrayTensor(
         data: AnyRef): Array[Tensor[Float]] = {
     data match {
       case ndArray: NDArray[_] =>
@@ -418,7 +418,7 @@ object PythonLoaderFeatureSet{
     }
   }
 
-  protected def ndArrayToTensor(ndArray: NDArray[_]): Tensor[Float] = {
+  private[zoo] def ndArrayToTensor(ndArray: NDArray[_]): Tensor[Float] = {
     val array = ndArray.asInstanceOf[NDArray[Array[_]]]
     val data = array.getData()
     data(0) match {