#BAEL-18260 Try and update the nd4j dependency to fix the build

This commit is contained in:
Alessio Stalla
2019-10-13 09:38:38 +02:00
parent 8775cdb106
commit 3a7c2ac8cc
2 changed files with 7 additions and 2 deletions

View File

@@ -14,6 +14,11 @@
</parent>
<dependencies>
<dependency>
<groupId>org.nd4j</groupId>
<artifactId>nd4j-api</artifactId>
<version>${dl4j.version}</version>
</dependency>
<dependency>
<groupId>org.nd4j</groupId>
<artifactId>nd4j-native-platform</artifactId>
@@ -43,7 +48,7 @@
</dependencies>
<properties>
<dl4j.version>0.9.1</dl4j.version>
<dl4j.version>1.0.0-beta5</dl4j.version>
</properties>
</project>

View File

@@ -165,4 +165,4 @@ public class MnistClassifier {
ModelSerializer.writeModel(model, modelPath, true);
logger.info("The MINIST model has been saved in {}", modelPath.getPath());
}
}
}