-
Notifications
You must be signed in to change notification settings - Fork 2k
Closed
tensorflow/tfjs-node
#243Labels
Description
TensorFlow.js version
1.0.2
Describe the problem or feature request
When the outputs on the right side of the sidebar exceed the size reserved for it in the template, the line wraps, and then the progress bar is redrawn at every step again.
There is a proposed fix for this in the library you use to draw the progress bar, but it would probably also be possible to work around it inside tensorflow by ensuring the name of the placeholder is padded so it is long enough to cover the expected size of its replacement.
Code to reproduce the bug / link to feature request
let input = tf.input({ shape: [max_len, recogized_letters.length] });
let $ = tf.layers.flatten().apply(input);
let $1 = tf.layers.dense({ name:"hi_this_is_the_first_model", units: 2}).apply($);
let $2 = tf.layers.dense({ name:"hi_this_is_the_second_model", units: 2}).apply($);
let model = tf.model({ inputs: [input], outputs: [ $1, $2 ] });
model.compile({
loss: 'meanSquaredError',
optimizer: 'adam',
metrics: ['acc']
});
await model.fit(xTrain, [yTrain,yTrain], {
epochs: 2,
batchSize: 1,
validationSplit: 0.2,
});