Merge pull request #4090 from deeplearning4j/sa_kerasfix

Allow setting WorkspaceMode for transfer learning as well

deeplearning4j-nn/src/main/java/org/deeplearning4j/nn/transferlearning/FineTuneConfiguration.java

@@ -67,6 +67,9 @@ public class FineTuneConfiguration {
     protected Integer tbpttFwdLength;
     protected Integer tbpttBackLength;
 
+    protected WorkspaceMode trainingWorkspaceMode;
+    protected WorkspaceMode inferenceWorkspaceMode;
+
     //Lombok builder. Note that the code below ADDS OR OVERRIDES the lombok implementation; the final builder class
     // is the composite of the lombok parts and the parts defined here
     //partial implementation to allow public no-arg constructor (lombok default is package private)
@@ -268,6 +271,10 @@ public NeuralNetConfiguration.Builder appliedNeuralNetConfigurationBuilder() {
             confBuilder.setGradientNormalization(gradientNormalization);
         if (gradientNormalizationThreshold != null)
             confBuilder.setGradientNormalizationThreshold(gradientNormalizationThreshold);
+        if (trainingWorkspaceMode != null)
+            confBuilder.trainingWorkspaceMode(trainingWorkspaceMode);
+        if (inferenceWorkspaceMode != null)
+            confBuilder.inferenceWorkspaceMode(inferenceWorkspaceMode);
         return confBuilder;
     }