Search in sources :

Example 1 with Heartbeat

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);
    }
}
Also used : Task(org.nd4j.linalg.heartbeat.reports.Task) Heartbeat(org.nd4j.linalg.heartbeat.Heartbeat)

Example 2 with Heartbeat

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);
    }
}
Also used : Heartbeat(org.nd4j.linalg.heartbeat.Heartbeat) Environment(org.nd4j.linalg.heartbeat.reports.Environment)

Example 3 with Heartbeat

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);
    }
}
Also used : Heartbeat(org.nd4j.linalg.heartbeat.Heartbeat) Environment(org.nd4j.linalg.heartbeat.reports.Environment)

Aggregations

Heartbeat (org.nd4j.linalg.heartbeat.Heartbeat)3 Environment (org.nd4j.linalg.heartbeat.reports.Environment)2 Task (org.nd4j.linalg.heartbeat.reports.Task)1