We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 26945e0 commit 4f3fbdfCopy full SHA for 4f3fbdf
src/NeuralNetwork/index.js
@@ -318,8 +318,8 @@ class NeuralNetwork {
318
*/
319
async trainInternal(options, whileTrainingCallback) {
320
// get batch size and epochs
321
- const batchSize = options.batchSize || this.config.batchSize;
322
- const epochs = options.epochs || this.config.epochs;
+ const batchSize = options.batchSize || this.config.training.batchSize;
+ const epochs = options.epochs || this.config.training.epochs;
323
324
// placeholder for whiletraining callback;
325
let whileTraining;
0 commit comments