org.deeplearning4j.parallelism.factory

Class DefaultTrainerContext

java.lang.Object

org.deeplearning4j.parallelism.factory.DefaultTrainerContext

All Implemented Interfaces:

TrainerContext

public class DefaultTrainerContext
extends java.lang.Object
implements TrainerContext

Creates DefaultTrainer instances for use with ParallelWrapper

Constructor Summary

Constructors

Constructor and Description
DefaultTrainerContext()

Method Summary

Modifier and Type	Method and Description
Trainer	create(int threadId, Model model, int rootDevice, boolean useMDS, ParallelWrapper wrapper) Create a Trainer based on the given parameters
void	init(Model model, java.lang.Object... args) Initialize the context

Methods inherited from class java.lang.Object

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

Constructor Detail

DefaultTrainerContext

public DefaultTrainerContext()

Method Detail

init

public void init(Model model,
                 java.lang.Object... args)

Initialize the context

Specified by:

init in interface TrainerContext

Parameters:

model -

args - the arguments to initialize with (maybe null)

create

public Trainer create(int threadId,
                      Model model,
                      int rootDevice,
                      boolean useMDS,
                      ParallelWrapper wrapper)

Create a Trainer based on the given parameters

Specified by:

create in interface TrainerContext

Parameters:

threadId - the thread id to use for this worker

model - the model to start the trainer with

rootDevice - the root device id

useMDS - whether to use the MagicQueue or not

wrapper - the wrapper instance to use with this trainer (this refernece is needed for coordination with the ParallelWrapper 's IterationListener

Returns:

the created training instance