Fix NPE bug in block.toString() function

Change-Id: I2aa8bc6c8d36d01cc00c02778907b040d782330c
deepjavalibrary · Jul 5, 2021 · 3815157 · 3815157
1 parent 7a7f41f
commit 3815157
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/api/src/main/java/ai/djl/nn/AbstractBlock.java b/api/src/main/java/ai/djl/nn/AbstractBlock.java
@@ -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;