Packages

class TrainSummary extends Summary

Train logger for tensorboard. Use optimize.setTrainSummary to enable train logger. Then the log will be saved to logDir/appName/train.

Linear Supertypes
Summary, AnyRef, Any
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. TrainSummary
  2. Summary
  3. AnyRef
  4. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Instance Constructors

  1. new TrainSummary(logDir: String, appName: String)

    logDir

    log dir.

    appName

    application Name.

Value Members

  1. final def !=(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  2. final def ##(): Int
    Definition Classes
    AnyRef → Any
  3. final def ==(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  4. def addHistogram[T](tag: String, value: Tensor[T], step: Long)(implicit arg0: ClassTag[T], ev: TensorNumeric[T]): TrainSummary.this.type

    Add a histogram summary.

    Add a histogram summary.

    tag

    tag name.

    value

    a tensor.

    step

    current step.

    returns

    this

    Definition Classes
    Summary
  5. def addScalar(tag: String, value: Float, step: Long): TrainSummary.this.type

    Add a scalar summary.

    Add a scalar summary.

    tag

    tag name.

    value

    tag value.

    step

    current step.

    returns

    this

    Definition Classes
    Summary
  6. final def asInstanceOf[T0]: T0
    Definition Classes
    Any
  7. def clone(): AnyRef
    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @native() @throws( ... )
  8. def close(): Unit

    Close this logger.

    Close this logger.

    Definition Classes
    Summary
  9. final def eq(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  10. def equals(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  11. def finalize(): Unit
    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  12. val folder: String
    Attributes
    protected
  13. final def getClass(): Class[_]
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  14. def getSummaryTrigger(tag: String): Option[Trigger]

    Get a trigger by tag name.

  15. def hashCode(): Int
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  16. final def isInstanceOf[T0]: Boolean
    Definition Classes
    Any
  17. final def ne(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  18. final def notify(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  19. final def notifyAll(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  20. def readScalar(tag: String): Array[(Long, Float, Double)]

    Read scalar values to an array of triple by tag name.

    Read scalar values to an array of triple by tag name. First element of the triple is step, second is value, third is wallClockTime.

    tag

    tag name. Supported tag names is "LearningRate", "Loss", "Throughput"

    returns

    an array of triple.

    Definition Classes
    TrainSummarySummary
  21. def setSummaryTrigger(tag: String, trigger: Trigger): TrainSummary.this.type

    Supported tag name are LearningRate, Loss, Throughput, Parameters.

    Supported tag name are LearningRate, Loss, Throughput, Parameters. Parameters contains weight, bias, gradWeight, gradBias, and some running status(eg. runningMean and runningVar in BatchNormalization).

    Notice: By default, we record LearningRate, Loss and Throughput each iteration, while recording parameters is disabled. The reason is getting parameters from workers is a heavy operation when the model is very big.

    tag

    tag name

    trigger

    trigger

  22. final def synchronized[T0](arg0: ⇒ T0): T0
    Definition Classes
    AnyRef
  23. def toString(): String
    Definition Classes
    AnyRef → Any
  24. final def wait(): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  25. final def wait(arg0: Long, arg1: Int): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  26. final def wait(arg0: Long): Unit
    Definition Classes
    AnyRef
    Annotations
    @native() @throws( ... )
  27. val writer: FileWriter
    Attributes
    protected
    Definition Classes
    TrainSummarySummary

Inherited from Summary

Inherited from AnyRef

Inherited from Any

Ungrouped