Method setModel

tfjs-layers/src/base_callbacks.ts:128–132

(model: Container)

  }

  setModel(model: Container): void {
    for (const callback of this.callbacks) {
      callback.setModel(model);
    }
  }

  /**
   * Called at the start of an epoch.