Class

com.intel.analytics.bigdl.example.languagemodel.Utils

TrainParams

Related Doc: package Utils

Permalink

case class TrainParams(dataFolder: String = "./", modelSnapshot: Option[String] = None, stateSnapshot: Option[String] = None, checkpoint: Option[String] = None, batchSize: Int = 20, learningRate: Double = 0.01, learningRateDecay: Double = 0.001, hiddenSize: Int = 200, vocabSize: Int = 10000, nEpochs: Int = 4, numLayers: Int = 2, numSteps: Int = 20, overWriteCheckpoint: Boolean = false, keepProb: Float = 2.0f, withTransformerModel: Boolean = false, optimizerVersion: Option[String] = None) extends Product with Serializable

Linear Supertypes
Serializable, Serializable, Product, Equals, AnyRef, Any
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. TrainParams
  2. Serializable
  3. Serializable
  4. Product
  5. Equals
  6. AnyRef
  7. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Instance Constructors

  1. new TrainParams(dataFolder: String = "./", modelSnapshot: Option[String] = None, stateSnapshot: Option[String] = None, checkpoint: Option[String] = None, batchSize: Int = 20, learningRate: Double = 0.01, learningRateDecay: Double = 0.001, hiddenSize: Int = 200, vocabSize: Int = 10000, nEpochs: Int = 4, numLayers: Int = 2, numSteps: Int = 20, overWriteCheckpoint: Boolean = false, keepProb: Float = 2.0f, withTransformerModel: Boolean = false, optimizerVersion: Option[String] = None)

    Permalink

Value Members

  1. final def !=(arg0: Any): Boolean

    Permalink
    Definition Classes
    AnyRef → Any

  2. final def ##(): Int

    Permalink
    Definition Classes
    AnyRef → Any

  3. final def ==(arg0: Any): Boolean

    Permalink
    Definition Classes
    AnyRef → Any

  4. final def asInstanceOf[T0]: T0

    Permalink
    Definition Classes
    Any

  5. val batchSize: Int

    Permalink

  6. val checkpoint: Option[String]

    Permalink

  7. def clone(): AnyRef

    Permalink
    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

  8. val dataFolder: String

    Permalink

  9. final def eq(arg0: AnyRef): Boolean

    Permalink
    Definition Classes
    AnyRef

  10. def finalize(): Unit

    Permalink
    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )

  11. final def getClass(): Class[_]

    Permalink
    Definition Classes
    AnyRef → Any

  12. val hiddenSize: Int

    Permalink

  13. final def isInstanceOf[T0]: Boolean

    Permalink
    Definition Classes
    Any

  14. val keepProb: Float

    Permalink

  15. val learningRate: Double

    Permalink

  16. val learningRateDecay: Double

    Permalink

  17. val modelSnapshot: Option[String]

    Permalink

  18. val nEpochs: Int

    Permalink

  19. final def ne(arg0: AnyRef): Boolean

    Permalink
    Definition Classes
    AnyRef

  20. final def notify(): Unit

    Permalink
    Definition Classes
    AnyRef

  21. final def notifyAll(): Unit

    Permalink
    Definition Classes
    AnyRef

  22. val numLayers: Int

    Permalink

  23. val numSteps: Int

    Permalink

  24. val optimizerVersion: Option[String]

    Permalink

  25. val overWriteCheckpoint: Boolean

    Permalink

  26. val stateSnapshot: Option[String]

    Permalink

  27. final def synchronized[T0](arg0: ⇒ T0): T0

    Permalink
    Definition Classes
    AnyRef

  28. val vocabSize: Int

    Permalink

  29. final def wait(): Unit

    Permalink
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

  30. final def wait(arg0: Long, arg1: Int): Unit

    Permalink
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

  31. final def wait(arg0: Long): Unit

    Permalink
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

  32. val withTransformerModel: Boolean

    Permalink

Inherited from Serializable

Inherited from Serializable

Inherited from Product

Inherited from Equals

Inherited from AnyRef

Inherited from Any

Ungrouped