HouseNumberDetection.

Signed-off-by: Robert Altena <[email protected]>

Author: RobAltena

dl4j-examples/src/main/java/org/deeplearning4j/examples/convolution/objectdetection/HouseNumberDetection.java

@@ -1,4 +1,4 @@
-/*******************************************************************************
+/* *****************************************************************************
  * Copyright (c) 2015-2019 Skymind, Inc.
  *
  * This program and the accompanying materials are made available under the
@@ -54,7 +54,6 @@
 import java.util.Random;
 
 import org.bytedeco.opencv.opencv_core.*;
-import org.bytedeco.opencv.opencv_imgproc.*;
 import static org.bytedeco.opencv.global.opencv_core.*;
 import static org.bytedeco.opencv.global.opencv_imgproc.*;
 
@@ -96,7 +95,6 @@ public static void main(String[] args) throws java.lang.Exception {
         int batchSize = 10;
         int nEpochs = 20;
         double learningRate = 1e-4;
-        double lrMomentum = 0.9;
 
         int seed = 123;
         Random rng = new Random(seed);
@@ -149,8 +147,8 @@ public static void main(String[] args) throws java.lang.Exception {
                     //.updater(new Nesterovs.Builder().learningRate(learningRate).momentum(lrMomentum).build())
                     .l2(0.00001)
                     .activation(Activation.IDENTITY)
-                    .trainingWorkspaceMode(WorkspaceMode.SEPARATE)
-                    .inferenceWorkspaceMode(WorkspaceMode.SEPARATE)
+                    .trainingWorkspaceMode(WorkspaceMode.ENABLED)
+                    .inferenceWorkspaceMode(WorkspaceMode.ENABLED)
                     .build();
 
             model = new TransferLearning.GraphBuilder(pretrained)