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

Small fix for clashing dependency (test scope only) (#214) 

Signed-off-by: AlexDBlack <blacka101@gmail.com>
master
Alex Black 2019-09-01 00:16:38 +10:00 committed by GitHub
parent 12d14efcf2
commit f00a7bb3f2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
deeplearning4j/deeplearning4j-cuda/src/test/java/org/deeplearning4j/ValidateCuDNN.java
View File

@ -248,7 +248,7 @@ public class ValidateCuDNN extends BaseDL4JTest {
Nd4j.getRandom().setSeed(12345);
INDArray features = Nd4j.rand(fShape);
INDArray labels = Nd4j.rand(lShape);
labels = Nd4j.exec(new IsMax(labels, 1));
labels = Nd4j.exec(new IsMax(labels, 1))[0].castTo(features.dataType());
List<CuDNNValidationUtil.TestCase> testCaseList = new ArrayList<>();
@ -256,7 +256,7 @@ public class ValidateCuDNN extends BaseDL4JTest {
for (int i = 0; i < 6; i++) {
INDArray f = Nd4j.rand(fShape);
INDArray l = Nd4j.rand(lShape);
Nd4j.exec(new IsMax(l, 1))[0];
l = Nd4j.exec(new IsMax(l, 1))[0].castTo(features.dataType());
dataSets.add(new DataSet(f, l));
}
DataSetIterator iter = new ExistingDataSetIterator(dataSets);

7
deeplearning4j/deeplearning4j-scaleout/spark/dl4j-spark/pom.xml
View File

@ -70,6 +70,13 @@
<artifactId>deeplearning4j-play_2.11</artifactId>
<version>${deeplearning4j.version}</version>
<scope>test</scope>
<exclusions>
<!-- To avoid clashing net.jpountz.lz4:lz4:1.3.0 and org.lz4:lz4-java:jar:1.4.0 - -->
<exclusion>
<groupId>net.jpountz.lz4</groupId>
<artifactId>lz4</artifactId>
</exclusion>
</exclusions>
</dependency>
<dependency>