org.deeplearning4j.spark.api

Class WorkerConfiguration

java.lang.Object

org.deeplearning4j.spark.api.WorkerConfiguration

All Implemented Interfaces:

java.io.Serializable

public class WorkerConfiguration
extends java.lang.Object
implements java.io.Serializable

A simple configuration object (common settings for workers)

See Also:

Serialized Form

Field Summary

Fields

Modifier and Type	Field and Description
protected int	batchSizePerWorker
protected boolean	collectTrainingStats
protected int	dataSetObjectSizeExamples
protected boolean	isGraphNetwork
protected int	maxBatchesPerWorker
protected int	prefetchNumBatches

Constructor Summary

Constructors

Constructor and Description
WorkerConfiguration()

Method Summary

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Detail

isGraphNetwork

protected final boolean isGraphNetwork

dataSetObjectSizeExamples

protected final int dataSetObjectSizeExamples

batchSizePerWorker

protected final int batchSizePerWorker

maxBatchesPerWorker

protected final int maxBatchesPerWorker

prefetchNumBatches

protected final int prefetchNumBatches

collectTrainingStats

protected final boolean collectTrainingStats

Constructor Detail

WorkerConfiguration

public WorkerConfiguration()