org.nd4j.parameterserver.distributed.training

Class TrainerProvider

java.lang.Object

org.nd4j.parameterserver.distributed.training.TrainerProvider

public class TrainerProvider
extends Object

Author:

raver119@gmail.com

Field Summary

Fields

Modifier and Type	Field and Description
protected Clipboard	clipboard
protected Storage	storage
protected Map<String,TrainingDriver<?>>	trainers
protected Transport	transport
protected VoidConfiguration	voidConfiguration

Method Summary

Modifier and Type	Method and Description
<T extends TrainingMessage> void	doTraining(T message)
static TrainerProvider	getInstance()
protected <T extends TrainingMessage> TrainingDriver<T>	getTrainer(T message)
void	init(VoidConfiguration voidConfiguration, Transport transport, Storage storage, Clipboard clipboard)
protected void	scanClasspath()

Methods inherited from class java.lang.Object

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

Field Detail

trainers

protected Map<String,TrainingDriver<?>> trainers

voidConfiguration

protected VoidConfiguration voidConfiguration

transport

protected Transport transport

clipboard

protected Clipboard clipboard

storage

protected Storage storage

Method Detail

getInstance

public static TrainerProvider getInstance()

scanClasspath

protected void scanClasspath()

init

public void init(@NonNull
                 VoidConfiguration voidConfiguration,
                 @NonNull
                 Transport transport,
                 @NonNull
                 Storage storage,
                 @NonNull
                 Clipboard clipboard)

getTrainer

protected <T extends TrainingMessage> TrainingDriver<T> getTrainer(T message)

doTraining

public <T extends TrainingMessage> void doTraining(T message)