Fix NPE bug in block.toString() function

Change-Id: I2aa8bc6c8d36d01cc00c02778907b040d782330c
deepjavalibrary · Jul 5, 2021 · 122c355 · 122c355
1 parent 7a7f41f
commit 122c355
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;