Skip to content

Commit

Permalink
Gradle: increase dl4j libs versions up to 1.0.0-M2 (#2)
Browse files Browse the repository at this point in the history
* BOT: increase dl4j libs versions

* BOT: add explicitly error logging

* BOT: Change dl4j version to 1.0.0-M2 and specify classifier "linux-arm64" for nd4j
  • Loading branch information
sbaldin authored Apr 24, 2022
1 parent 7d93df7 commit 5883965
Show file tree
Hide file tree
Showing 2 changed files with 11 additions and 3 deletions.
13 changes: 10 additions & 3 deletions build.gradle
Original file line number Diff line number Diff line change
Expand Up @@ -83,9 +83,16 @@ dependencies {

implementation 'com.vdurmont:emoji-java:5.1.1'

implementation "org.deeplearning4j:deeplearning4j-core:1.0.0-beta7"
implementation "org.deeplearning4j:deeplearning4j-zoo:1.0.0-beta7"
implementation "org.nd4j:nd4j-native-platform:1.0.0-beta7"
/*
implementation "org.deeplearning4j:deeplearning4j-core:1.0.0-M2"
implementation "org.deeplearning4j:deeplearning4j-zoo:1.0.0-M2"*/
implementation group: 'org.nd4j', name: 'nd4j-native', version: '1.0.0-M2'
implementation group: 'org.nd4j', name: 'nd4j-native', version: '1.0.0-M2', classifier: "linux-arm64"
implementation group: 'org.deeplearning4j', name: 'deeplearning4j-core', version: '1.0.0-M2'
implementation group: 'org.deeplearning4j', name: 'deeplearning4j-zoo', version: '1.0.0-M2'

implementation group: 'org.bytedeco', name: 'openblas', version: '0.3.5-1.5'
implementation group: 'org.bytedeco', name: 'openblas', version: '0.3.5-1.5', classifier: "linux-arm64"

testImplementation "io.mockk:mockk:${mockkversion}"
// Use the Kotlin test library.
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,7 @@ class BirdClassifier(conf: CnnConf) {
return try {
FileInputStream(File(conf.modelPath))
} catch (e: Exception) {
log.error("Error has occurred during model loading!", e)
Thread.currentThread().contextClassLoader.getResourceAsStream(conf.modelPath)!!
}
}
Expand Down

0 comments on commit 5883965

Please sign in to comment.