@Deprecated public class SilentTrainingDriver extends Object implements TrainingDriver<SilentUpdatesMessage>
Modifier and Type | Field and Description |
---|---|
protected GradientsAccumulator |
accumulator
Deprecated.
|
protected AtomicBoolean |
bypassMode
Deprecated.
|
protected Clipboard |
clipboard
Deprecated.
|
protected AtomicLong |
denseCounter
Deprecated.
|
protected AtomicBoolean |
hasSomething
Deprecated.
|
protected INDArray |
params
Deprecated.
|
protected AtomicLong |
sparseCounter
Deprecated.
|
protected StepFunction |
stepFunction
Deprecated.
|
protected Storage |
storage
Deprecated.
|
protected Transport |
transport
Deprecated.
|
protected INDArray |
updates
Deprecated.
|
protected IndexedTail |
updatesBuffer
Deprecated.
|
protected AtomicLong |
updatesCount
Deprecated.
|
protected VoidConfiguration |
voidConfiguration
Deprecated.
|
Constructor and Description |
---|
SilentTrainingDriver(@NonNull GradientsAccumulator accumulator)
Deprecated.
|
SilentTrainingDriver(@NonNull INDArray params,
@NonNull StepFunction stepFunction)
Deprecated.
|
Modifier and Type | Method and Description |
---|---|
void |
addCompletionHook(long originatorId,
long frameId,
long messageId)
Deprecated.
|
void |
aggregationFinished(VoidAggregation aggregation)
Deprecated.
|
void |
bypassMode(boolean reallyBypass)
Deprecated.
|
void |
finishTraining(long originatorId,
long taskId)
Deprecated.
This method is used on Master only, applies buffered updates to params
|
IndexedTail |
getUpdatesBuffer()
Deprecated.
This method is viable only at Spark Workers, Master node will always have empty buffer here by design
|
void |
init(@NonNull VoidConfiguration voidConfiguration,
@NonNull Transport transport,
Storage storage,
Clipboard clipboard)
Deprecated.
|
void |
pickTraining(SilentUpdatesMessage message)
Deprecated.
|
void |
startTraining(SilentUpdatesMessage message)
Deprecated.
|
String |
targetMessageClass()
Deprecated.
|
protected transient INDArray params
protected transient INDArray updates
protected transient StepFunction stepFunction
protected transient GradientsAccumulator accumulator
protected transient VoidConfiguration voidConfiguration
protected transient Transport transport
protected transient AtomicLong updatesCount
protected transient AtomicBoolean hasSomething
protected transient AtomicBoolean bypassMode
protected transient AtomicLong denseCounter
protected transient AtomicLong sparseCounter
protected transient IndexedTail updatesBuffer
protected transient Storage storage
protected transient Clipboard clipboard
public SilentTrainingDriver(@NonNull @NonNull GradientsAccumulator accumulator)
public SilentTrainingDriver(@NonNull @NonNull INDArray params, @NonNull @NonNull StepFunction stepFunction)
public IndexedTail getUpdatesBuffer()
public void init(@NonNull @NonNull VoidConfiguration voidConfiguration, @NonNull @NonNull Transport transport, Storage storage, Clipboard clipboard)
init
in interface TrainingDriver<SilentUpdatesMessage>
public void bypassMode(boolean reallyBypass)
public void startTraining(SilentUpdatesMessage message)
startTraining
in interface TrainingDriver<SilentUpdatesMessage>
public void pickTraining(SilentUpdatesMessage message)
pickTraining
in interface TrainingDriver<SilentUpdatesMessage>
public void aggregationFinished(VoidAggregation aggregation)
aggregationFinished
in interface TrainingDriver<SilentUpdatesMessage>
public void finishTraining(long originatorId, long taskId)
finishTraining
in interface TrainingDriver<SilentUpdatesMessage>
originatorId
- taskId
- public void addCompletionHook(long originatorId, long frameId, long messageId)
addCompletionHook
in interface TrainingDriver<SilentUpdatesMessage>
public String targetMessageClass()
targetMessageClass
in interface TrainingDriver<SilentUpdatesMessage>
Copyright © 2020. All rights reserved.