Method onBatchEnd

tfjs-layers/src/base_callbacks.ts:443–456

(batch: number, logs?: UnresolvedLogs)

  }

  override async onBatchEnd(batch: number, logs?: UnresolvedLogs):
      Promise<void> {
    const ps: Array<void|Promise<void>> = [];
    if (this.batchEnd != null) {
      await resolveScalarsInLogs(logs);
      ps.push(this.batchEnd(batch, logs as Logs));
    }
    if (this.yieldEvery === 'batch') {
      ps.push(this.nextFrameFunc());
    } else if (util.isNumber(this.yieldEvery)) {
      ps.push(this.maybeWait(this.currentEpoch, batch, logs));
    }
    await Promise.all(ps);
  }

  override async onTrainBegin(logs?: UnresolvedLogs): Promise<void> {
    if (this.trainBegin != null) {