use of org.nd4j.linalg.heartbeat.Heartbeat in project deeplearning4j by deeplearning4j.
the class WordVectorsImpl method update.
protected void update(Environment env, Event event) {
if (!initDone) {
initDone = true;
Heartbeat heartbeat = Heartbeat.getInstance();
Task task = new Task();
task.setNumFeatures(layerSize);
if (vocab != null)
task.setNumSamples(vocab.numWords());
task.setNetworkType(Task.NetworkType.DenseNetwork);
task.setArchitectureType(Task.ArchitectureType.WORDVECTORS);
heartbeat.reportEvent(event, env, task);
}
}
use of org.nd4j.linalg.heartbeat.Heartbeat in project deeplearning4j by deeplearning4j.
the class ComputationGraph method update.
private void update(Task task) {
if (!initDone) {
initDone = true;
Heartbeat heartbeat = Heartbeat.getInstance();
task = ModelSerializer.taskByModel(this);
Environment env = EnvironmentUtils.buildEnvironment();
heartbeat.reportEvent(Event.STANDALONE, env, task);
}
}
use of org.nd4j.linalg.heartbeat.Heartbeat in project deeplearning4j by deeplearning4j.
the class MultiLayerNetwork method update.
private void update(Task task) {
if (!initDone) {
initDone = true;
Heartbeat heartbeat = Heartbeat.getInstance();
task = ModelSerializer.taskByModel(this);
Environment env = EnvironmentUtils.buildEnvironment();
heartbeat.reportEvent(Event.STANDALONE, env, task);
}
}
Aggregations