Package

com.intel.analytics.bigdl.models

utils

Permalink

package utils

Visibility
  1. Public
  2. All

Type Members

  1. case class DistriOptimizerPerfParam(batchSize: Int = 128, maxEpoch: Int = 5, dataType: String = "float", module: String = "inception_v1", inputData: String = "random") extends Product with Serializable

    Permalink

    The parameters of a distributed optimizer

    The parameters of a distributed optimizer

    batchSize

    batch size

    maxEpoch

    how many epochs to run

    dataType

    data type (double / float)

    module

    module name

    inputData

    inputData input data type (constant / random)

  2. case class LocalOptimizerPerfParam(batchSize: Int = 128, coreNumber: Int = ..., iteration: Int = 50, dataType: String = "float", module: String = "inception_v1", inputData: String = "random") extends Product with Serializable

    Permalink

    Local Optimizer Performance Parameters

    Local Optimizer Performance Parameters

    batchSize

    batch size

    coreNumber

    core number

    iteration

    how many iterations to run

    dataType

    data type (double / float)

    module

    module name

    inputData

    input data type (constant / random)

  3. trait ModelBroadcast[T] extends Serializable

    Permalink

    ModelBroadcast is used to broadcast model

  4. trait ModelBroadcastFactory extends AnyRef

    Permalink

    Create a ModelBroadcast.

    Create a ModelBroadcast. User can override how model is broadcast by extending this class to create a customized ModelBroadcast. To enable customized broadcast factory, user need to specify its full class name via system property bigdl.ModelBroadcastFactory.

Value Members

  1. object COCOSeqFileGenerator

    Permalink
  2. object CachedModels

    Permalink
  3. object DistriOptimizerPerf

    Permalink
  4. object ImageNetSeqFileGenerator

    Permalink
  5. object LocalOptimizerPerf

    Permalink
  6. object ModelBroadcast extends Serializable

    Permalink
  7. object ModelInfo extends Serializable

    Permalink

Ungrouped