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

Adding cuDNN support 

Signed-off-by: brian <brian@brutex.de>
master
Brian Rosenberger 2023-03-10 16:32:41 +01:00
parent bf10564be7
commit 8d31caffbe
3 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
brutex-extended-tests/src/test/java/net/brutex/gan/App.java
View File

@ -167,7 +167,7 @@ public class App {
trainData.reset();
int j = 0;
for (int i = 0; i < 10; i++) {
for (int i = 0; i < 20; i++) {
while (trainData.hasNext()) {
j++;

6
cavis-dnn/cavis-dnn-api/src/main/java/org/nd4j/linalg/factory/Nd4j.java
View File

@ -4877,7 +4877,7 @@ public class Nd4j {
* Create an ndarray of zeros
*
* @param shape the shape of the array
* @return an ndarray with ones filled in
* @return an ndarray with zeros filled in
*/
public static INDArray zeros(int[] shape, char order) {
checkShapeValues(shape);
@ -4896,7 +4896,7 @@ public class Nd4j {
* Create an ndarray of zeros
*
* @param shape the shape of the array
* @return an ndarray with ones filled in
* @return an ndarray with zeros filled in
*/
public static INDArray zeros(@NonNull int... shape) {
return Nd4j.create(shape);
@ -4907,7 +4907,7 @@ public class Nd4j {
* Create an ndarray of zeros
*
* @param shape the shape of the array
* @return an ndarray with ones filled in
* @return an ndarray with zeros filled in
*/
public static INDArray zeros(@NonNull long... shape) {
return Nd4j.create(shape);

3
cavis-dnn/cavis-dnn-common/src/main/java/org/deeplearning4j/common/config/DL4JClassLoading.java
View File

@ -105,7 +105,8 @@ public class DL4JClassLoading {
throw new RuntimeException(instantiationException);
} catch (InvocationTargetException instantiationException) {
log.error(String.format("InvocationTargetException was '%s'.", instantiationException.getTargetException().getMessage()), instantiationException);
log.error(String.format("---------- ----------- ---------- \nInvocationTargetException was '%s'.", instantiationException.getTargetException().getMessage()), instantiationException);
log.error(String.format("java.library.path was '%s'\n---------- ---------- ----------", System.getProperty("java.library.path")));
throw new RuntimeException(instantiationException);
}
}