Brutex
/
cavis
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

MultiLayerConfiguration should be ComputationGraphConfiguration for validating a cg model conf (#240)

master
Shams Ul Azeem 2020-02-14 06:06:00 +05:00 committed by GitHub
parent c34790f932
commit f165160edb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
deeplearning4j/deeplearning4j-nn/src/main/java/org/deeplearning4j/util/DL4JModelValidator.java
View File

@ -3,6 +3,7 @@ package org.deeplearning4j.util;
import lombok.NonNull; import lombok.NonNull;
import org.apache.commons.io.IOUtils; import org.apache.commons.io.IOUtils;
import org.deeplearning4j.nn.api.Model; import org.deeplearning4j.nn.api.Model;
import org.deeplearning4j.nn.conf.ComputationGraphConfiguration;
import org.deeplearning4j.nn.conf.MultiLayerConfiguration; import org.deeplearning4j.nn.conf.MultiLayerConfiguration;
import org.deeplearning4j.nn.graph.ComputationGraph; import org.deeplearning4j.nn.graph.ComputationGraph;
import org.deeplearning4j.nn.multilayer.MultiLayerNetwork; import org.deeplearning4j.nn.multilayer.MultiLayerNetwork;
@ -121,7 +122,7 @@ public class DL4JModelValidator {
} }
try{ try{
MultiLayerConfiguration.fromJson(config); ComputationGraphConfiguration.fromJson(config);
} catch (Throwable t){ } catch (Throwable t){
return ValidationResult.builder() return ValidationResult.builder()
.formatType("ComputationGraph") .formatType("ComputationGraph")