Fix NPE bug in block.toString() function (#1076)

Change-Id: I2aa8bc6c8d36d01cc00c02778907b040d782330c
deepjavalibrary · Jul 6, 2021 · 06f76dd · 06f76dd
1 parent 7a7f41f
commit 06f76dd
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/api/src/main/java/ai/djl/nn/AbstractBlock.java b/api/src/main/java/ai/djl/nn/AbstractBlock.java
@@ -115,7 +115,7 @@ public final NDList forward(
             boolean training,
             PairList<String, Object> params) {
         NDManager paramsManager = parameterStore.getManager();
-        if (!isInitialized()) {
+        if (training && !isInitialized()) {
             initialize(paramsManager, DataType.FLOAT32, inputs.getShapes());
         }
         return forwardInternal(parameterStore, inputs, training, params);
@@ -330,6 +330,9 @@ protected void prepare(Shape[] inputShapes) {}
     /** {@inheritDoc} */
     @Override
     public boolean isInitialized() {
+        if (inputShapes == null) {
+            return false;
+        }
         for (Parameter param : getParameters().values()) {
             if (!param.isInitialized()) {
                 return false;