From bbbc3ba0bf924006e64fd3d37150fd1395f77a91 Mon Sep 17 00:00:00 2001 From: Na Li Date: Tue, 18 Feb 2020 16:50:54 -0800 Subject: [PATCH 1/2] Clarify usage of epochs and initalEpochs --- tfjs-layers/src/engine/training_tensors.ts | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/tfjs-layers/src/engine/training_tensors.ts b/tfjs-layers/src/engine/training_tensors.ts index 57dfc88cbe4..812265e7e6e 100644 --- a/tfjs-layers/src/engine/training_tensors.ts +++ b/tfjs-layers/src/engine/training_tensors.ts @@ -32,7 +32,12 @@ export interface ModelFitArgs { */ batchSize?: number; - /** The number of times to iterate over the training data arrays. */ + /** + * The number of times to iterate over the training data arrays. + * Note that when used with `initialEpoch`, epochs is the index of the + * "final epoch". The model is not trained for a number of iterations + * given by epochs, but merely until the epoch of index epochs is reached. + */ epochs?: number; /** From 2bc3683a38f372f3ac196f36556cd814fcbbf054 Mon Sep 17 00:00:00 2001 From: Na Li Date: Tue, 18 Feb 2020 16:51:40 -0800 Subject: [PATCH 2/2] fix format --- tfjs-layers/src/engine/training_tensors.ts | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tfjs-layers/src/engine/training_tensors.ts b/tfjs-layers/src/engine/training_tensors.ts index 812265e7e6e..b5dab627aaa 100644 --- a/tfjs-layers/src/engine/training_tensors.ts +++ b/tfjs-layers/src/engine/training_tensors.ts @@ -34,9 +34,9 @@ export interface ModelFitArgs { /** * The number of times to iterate over the training data arrays. - * Note that when used with `initialEpoch`, epochs is the index of the - * "final epoch". The model is not trained for a number of iterations - * given by epochs, but merely until the epoch of index epochs is reached. + * Note that when used with `initialEpoch`, epochs is the index of the + * "final epoch". The model is not trained for a number of iterations + * given by epochs, but merely until the epoch of index epochs is reached. */ epochs?: number;