Class BaseStatsListener
- java.lang.Object
-
- org.deeplearning4j.ui.model.stats.BaseStatsListener
-
- All Implemented Interfaces:
Serializable,Cloneable,RoutingIterationListener,org.deeplearning4j.optimize.api.TrainingListener
- Direct Known Subclasses:
J7StatsListener,StatsListener
public abstract class BaseStatsListener extends Object implements RoutingIterationListener
- See Also:
- Serialized Form
-
-
Constructor Summary
Constructors Constructor Description BaseStatsListener(StatsStorageRouter router)Create a StatsListener with network information collected at every iteration.BaseStatsListener(StatsStorageRouter router, int listenerFrequency)Create a StatsListener with network information collected every n >= 1 time stepsBaseStatsListener(StatsStorageRouter router, StatsInitializationConfiguration initConfig, StatsUpdateConfiguration updateConfig, String sessionID, String workerID)
-
Method Summary
All Methods Instance Methods Abstract Methods Concrete Methods Modifier and Type Method Description abstract BaseStatsListenerclone()StatsInitializationConfigurationgetInitConfig()abstract StatsInitializationReportgetNewInitializationReport()abstract StatsReportgetNewStatsReport()abstract StorageMetaDatagetNewStorageMetaData(long initTime, String sessionID, String workerID)StringgetSessionID()StatsStorageRoutergetStorageRouter()StatsUpdateConfigurationgetUpdateConfig()StringgetWorkerID()voiditerationDone(org.deeplearning4j.nn.api.Model model, int iteration, int epoch)voidonBackwardPass(org.deeplearning4j.nn.api.Model model)voidonEpochEnd(org.deeplearning4j.nn.api.Model model)voidonEpochStart(org.deeplearning4j.nn.api.Model model)voidonForwardPass(org.deeplearning4j.nn.api.Model model, List<INDArray> activations)voidonForwardPass(org.deeplearning4j.nn.api.Model model, Map<String,INDArray> activations)voidonGradientCalculation(org.deeplearning4j.nn.api.Model model)voidsetSessionID(String sessionID)voidsetStorageRouter(StatsStorageRouter router)voidsetUpdateConfig(StatsUpdateConfiguration newConfig)voidsetWorkerID(String workerID)
-
-
-
Field Detail
-
TYPE_ID
public static final String TYPE_ID
- See Also:
- Constant Field Values
-
-
Constructor Detail
-
BaseStatsListener
public BaseStatsListener(StatsStorageRouter router)
Create a StatsListener with network information collected at every iteration.- Parameters:
router- Where/how to store the calculated stats. For example,InMemoryStatsStorageorFileStatsStorage
-
BaseStatsListener
public BaseStatsListener(StatsStorageRouter router, int listenerFrequency)
Create a StatsListener with network information collected every n >= 1 time steps- Parameters:
router- Where/how to store the calculated stats. For example,InMemoryStatsStorageorFileStatsStoragelistenerFrequency- Frequency with which to collect stats information
-
BaseStatsListener
public BaseStatsListener(StatsStorageRouter router, StatsInitializationConfiguration initConfig, StatsUpdateConfiguration updateConfig, String sessionID, String workerID)
-
-
Method Detail
-
getNewInitializationReport
public abstract StatsInitializationReport getNewInitializationReport()
-
getNewStatsReport
public abstract StatsReport getNewStatsReport()
-
getNewStorageMetaData
public abstract StorageMetaData getNewStorageMetaData(long initTime, String sessionID, String workerID)
-
getInitConfig
public StatsInitializationConfiguration getInitConfig()
-
getUpdateConfig
public StatsUpdateConfiguration getUpdateConfig()
-
setUpdateConfig
public void setUpdateConfig(StatsUpdateConfiguration newConfig)
-
setStorageRouter
public void setStorageRouter(StatsStorageRouter router)
- Specified by:
setStorageRouterin interfaceRoutingIterationListener
-
getStorageRouter
public StatsStorageRouter getStorageRouter()
- Specified by:
getStorageRouterin interfaceRoutingIterationListener
-
setWorkerID
public void setWorkerID(String workerID)
- Specified by:
setWorkerIDin interfaceRoutingIterationListener
-
getWorkerID
public String getWorkerID()
- Specified by:
getWorkerIDin interfaceRoutingIterationListener
-
setSessionID
public void setSessionID(String sessionID)
- Specified by:
setSessionIDin interfaceRoutingIterationListener
-
getSessionID
public String getSessionID()
- Specified by:
getSessionIDin interfaceRoutingIterationListener
-
onEpochStart
public void onEpochStart(org.deeplearning4j.nn.api.Model model)
- Specified by:
onEpochStartin interfaceorg.deeplearning4j.optimize.api.TrainingListener
-
onEpochEnd
public void onEpochEnd(org.deeplearning4j.nn.api.Model model)
- Specified by:
onEpochEndin interfaceorg.deeplearning4j.optimize.api.TrainingListener
-
onForwardPass
public void onForwardPass(org.deeplearning4j.nn.api.Model model, List<INDArray> activations)- Specified by:
onForwardPassin interfaceorg.deeplearning4j.optimize.api.TrainingListener
-
onForwardPass
public void onForwardPass(org.deeplearning4j.nn.api.Model model, Map<String,INDArray> activations)- Specified by:
onForwardPassin interfaceorg.deeplearning4j.optimize.api.TrainingListener
-
onGradientCalculation
public void onGradientCalculation(org.deeplearning4j.nn.api.Model model)
- Specified by:
onGradientCalculationin interfaceorg.deeplearning4j.optimize.api.TrainingListener
-
onBackwardPass
public void onBackwardPass(org.deeplearning4j.nn.api.Model model)
- Specified by:
onBackwardPassin interfaceorg.deeplearning4j.optimize.api.TrainingListener
-
iterationDone
public void iterationDone(org.deeplearning4j.nn.api.Model model, int iteration, int epoch)- Specified by:
iterationDonein interfaceorg.deeplearning4j.optimize.api.TrainingListener
-
clone
public abstract BaseStatsListener clone()
- Specified by:
clonein interfaceRoutingIterationListener- Overrides:
clonein classObject
-
-