Skip to content

Commit a6695de

Browse files
committed
Merge branch 'TobiasNickel-fix-usage-of-default-training-options' into release
2 parents 26945e0 + 4f3fbdf commit a6695de

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/NeuralNetwork/index.js

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -318,8 +318,8 @@ class NeuralNetwork {
318318
*/
319319
async trainInternal(options, whileTrainingCallback) {
320320
// get batch size and epochs
321-
const batchSize = options.batchSize || this.config.batchSize;
322-
const epochs = options.epochs || this.config.epochs;
321+
const batchSize = options.batchSize || this.config.training.batchSize;
322+
const epochs = options.epochs || this.config.training.epochs;
323323

324324
// placeholder for whiletraining callback;
325325
let whileTraining;

0 commit comments

Comments
 (0)