public abstract class BaseStatsListener extends Object implements RoutingIterationListener
Serves as a base for different ways of storing the collected data
| Constructor and 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 steps
|
BaseStatsListener(StatsStorageRouter router,
StatsInitializationConfiguration initConfig,
StatsUpdateConfiguration updateConfig,
String sessionID,
String workerID) |
| Modifier and Type | Method and Description |
|---|---|
abstract BaseStatsListener |
clone() |
StatsInitializationConfiguration |
getInitConfig() |
abstract StatsInitializationReport |
getNewInitializationReport() |
abstract StatsReport |
getNewStatsReport() |
abstract StorageMetaData |
getNewStorageMetaData(long initTime,
String sessionID,
String workerID) |
String |
getSessionID() |
StatsStorageRouter |
getStorageRouter() |
StatsUpdateConfiguration |
getUpdateConfig() |
String |
getWorkerID() |
void |
iterationDone(org.deeplearning4j.nn.api.Model model,
int iteration,
int epoch) |
void |
onBackwardPass(org.deeplearning4j.nn.api.Model model) |
void |
onEpochEnd(org.deeplearning4j.nn.api.Model model) |
void |
onEpochStart(org.deeplearning4j.nn.api.Model model) |
void |
onForwardPass(org.deeplearning4j.nn.api.Model model,
List<INDArray> activations) |
void |
onForwardPass(org.deeplearning4j.nn.api.Model model,
Map<String,INDArray> activations) |
void |
onGradientCalculation(org.deeplearning4j.nn.api.Model model) |
void |
setSessionID(String sessionID) |
void |
setStorageRouter(StatsStorageRouter router) |
void |
setUpdateConfig(StatsUpdateConfiguration newConfig) |
void |
setWorkerID(String workerID) |
public static final String TYPE_ID
public BaseStatsListener(StatsStorageRouter router)
router - Where/how to store the calculated stats. For example, InMemoryStatsStorage or
FileStatsStoragepublic BaseStatsListener(StatsStorageRouter router, int listenerFrequency)
router - Where/how to store the calculated stats. For example, InMemoryStatsStorage or
FileStatsStoragelistenerFrequency - Frequency with which to collect stats informationpublic BaseStatsListener(StatsStorageRouter router, StatsInitializationConfiguration initConfig, StatsUpdateConfiguration updateConfig, String sessionID, String workerID)
public abstract StatsInitializationReport getNewInitializationReport()
public abstract StatsReport getNewStatsReport()
public abstract StorageMetaData getNewStorageMetaData(long initTime, String sessionID, String workerID)
public StatsInitializationConfiguration getInitConfig()
public StatsUpdateConfiguration getUpdateConfig()
public void setUpdateConfig(StatsUpdateConfiguration newConfig)
public void setStorageRouter(StatsStorageRouter router)
setStorageRouter in interface RoutingIterationListenerpublic StatsStorageRouter getStorageRouter()
getStorageRouter in interface RoutingIterationListenerpublic void setWorkerID(String workerID)
setWorkerID in interface RoutingIterationListenerpublic String getWorkerID()
getWorkerID in interface RoutingIterationListenerpublic void setSessionID(String sessionID)
setSessionID in interface RoutingIterationListenerpublic String getSessionID()
getSessionID in interface RoutingIterationListenerpublic void onEpochStart(org.deeplearning4j.nn.api.Model model)
onEpochStart in interface org.deeplearning4j.optimize.api.TrainingListenerpublic void onEpochEnd(org.deeplearning4j.nn.api.Model model)
onEpochEnd in interface org.deeplearning4j.optimize.api.TrainingListenerpublic void onForwardPass(org.deeplearning4j.nn.api.Model model,
List<INDArray> activations)
onForwardPass in interface org.deeplearning4j.optimize.api.TrainingListenerpublic void onForwardPass(org.deeplearning4j.nn.api.Model model,
Map<String,INDArray> activations)
onForwardPass in interface org.deeplearning4j.optimize.api.TrainingListenerpublic void onGradientCalculation(org.deeplearning4j.nn.api.Model model)
onGradientCalculation in interface org.deeplearning4j.optimize.api.TrainingListenerpublic void onBackwardPass(org.deeplearning4j.nn.api.Model model)
onBackwardPass in interface org.deeplearning4j.optimize.api.TrainingListenerpublic void iterationDone(org.deeplearning4j.nn.api.Model model,
int iteration,
int epoch)
iterationDone in interface org.deeplearning4j.optimize.api.TrainingListenerpublic abstract BaseStatsListener clone()
clone in interface RoutingIterationListenerclone in class ObjectCopyright © 2019. All rights reserved.