object Module
- Alphabetic
- By Inheritance
- Module
- AnyRef
- Any
- Hide All
- Show All
- Public
- All
Value Members
-
final
def
!=(arg0: Any): Boolean
- Definition Classes
- AnyRef → Any
-
final
def
##(): Int
- Definition Classes
- AnyRef → Any
-
final
def
==(arg0: Any): Boolean
- Definition Classes
- AnyRef → Any
-
final
def
asInstanceOf[T0]: T0
- Definition Classes
- Any
-
def
clone(): AnyRef
- Attributes
- protected[java.lang]
- Definition Classes
- AnyRef
- Annotations
- @native() @throws( ... )
-
final
def
eq(arg0: AnyRef): Boolean
- Definition Classes
- AnyRef
-
def
equals(arg0: Any): Boolean
- Definition Classes
- AnyRef → Any
-
def
finalize(): Unit
- Attributes
- protected[java.lang]
- Definition Classes
- AnyRef
- Annotations
- @throws( classOf[java.lang.Throwable] )
- def flatten[T](parameters: Array[Tensor[T]])(implicit arg0: ClassTag[T], ev: TensorNumeric[T]): Tensor[T]
-
final
def
getClass(): Class[_]
- Definition Classes
- AnyRef → Any
- Annotations
- @native()
-
def
hashCode(): Int
- Definition Classes
- AnyRef → Any
- Annotations
- @native()
- def isCompact[T](parameters: Array[Tensor[T]])(implicit arg0: ClassTag[T], ev: TensorNumeric[T]): Tensor[T]
-
final
def
isInstanceOf[T0]: Boolean
- Definition Classes
- Any
-
def
loadCaffeModel[T](defPath: String, modelPath: String)(implicit arg0: ClassTag[T], ev: TensorNumeric[T]): AbstractModule[Activity, Activity, T]
Loaf caffe trained model from prototxt and weight files
Loaf caffe trained model from prototxt and weight files
- defPath
caffe model definition file path
- modelPath
caffe model binary file containing weight and bias
-
def
loadModule[T](path: String, weightPath: String = null)(implicit arg0: ClassTag[T], ev: TensorNumeric[T]): AbstractModule[Activity, Activity, T]
Load model from path.
Load model from path.
- T
numeric type
- path
path to save module, local file system, HDFS and Amazon S3 is supported. HDFS path should be like "hdfs://[host]:[port]/xxx" Amazon S3 path should be like "s3a://bucket/xxx"
- weightPath
: where weight is stored
- returns
model loaded from path
-
def
loadTF[T](graphFile: String, inputs: Seq[String], outputs: Seq[String], byteOrder: ByteOrder = ByteOrder.LITTLE_ENDIAN, binFile: Option[String] = None, generatedBackward: Boolean = true)(implicit arg0: ClassTag[T], ev: TensorNumeric[T]): Module[T]
Load tensorflow model from its saved protobuf file.
Load tensorflow model from its saved protobuf file.
- graphFile
where is the protobuf model file
- inputs
input node names
- outputs
output node names, the output tensor order is same with the node order
- byteOrder
byte order in the tensorflow file. The default value is little endian
- binFile
where is the model variable file
- generatedBackward
if generate backward graph
- returns
BigDL model
- def loadTorch[T](path: String)(implicit arg0: ClassTag[T]): AbstractModule[Activity, Activity, T]
-
final
def
ne(arg0: AnyRef): Boolean
- Definition Classes
- AnyRef
-
final
def
notify(): Unit
- Definition Classes
- AnyRef
- Annotations
- @native()
-
final
def
notifyAll(): Unit
- Definition Classes
- AnyRef
- Annotations
- @native()
-
final
def
synchronized[T0](arg0: ⇒ T0): T0
- Definition Classes
- AnyRef
-
def
tensorflowCheckpoints[T](graphFile: String, binFile: String, byteOrder: ByteOrder = ByteOrder.LITTLE_ENDIAN)(implicit arg0: ClassTag[T], ev: TensorNumeric[T]): Session[T]
Load tensorflow checkpoints
-
def
toString(): String
- Definition Classes
- AnyRef → Any
-
final
def
wait(): Unit
- Definition Classes
- AnyRef
- Annotations
- @throws( ... )
-
final
def
wait(arg0: Long, arg1: Int): Unit
- Definition Classes
- AnyRef
- Annotations
- @throws( ... )
-
final
def
wait(arg0: Long): Unit
- Definition Classes
- AnyRef
- Annotations
- @native() @throws( ... )
Deprecated Value Members
-
def
load[T](path: String)(implicit arg0: ClassTag[T]): AbstractModule[Activity, Activity, T]
Load model from path.
Load model from path.
- T
numeric type
- path
path to save module, local file system, HDFS and Amazon S3 is supported. HDFS path should be like "hdfs://[host]:[port]/xxx" Amazon S3 path should be like "s3a://bucket/xxx"
- returns
model loaded from path
- Annotations
- @deprecated
- Deprecated
(Since version 0.3) Java based serialization not recommended any more, please use loadModule instead
-
def
loadCaffe[T](model: AbstractModule[Activity, Activity, T], defPath: String, modelPath: String, matchAll: Boolean = true)(implicit arg0: ClassTag[T], ev: TensorNumeric[T]): AbstractModule[Activity, Activity, T]
- Annotations
- @deprecated
- Deprecated
(Since version 0.2) Please try to use the loadCaffeModel API