Demos/film-poster-genres
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

Improve logging

Browse source
This commit is contained in:
Starbeamrainbowlabs 2021-02-12 22:09:15 +00:00
parent 10138cd310
commit a5258f7460
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 1 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
src/lib/model.mjs
View file

@ -3,6 +3,7 @@
import tf from '@tensorflow/tfjs-node'; import tf from '@tensorflow/tfjs-node';
export default function(cats_count, image_size, use_crossentropy = false, activation = "relu") { export default function(cats_count, image_size, use_crossentropy = false, activation = "relu") {
console.error(`>>> Creating new model (activation = ${activation})`);
let model = tf.sequential(); let model = tf.sequential();
model.add(tf.layers.conv2d({ model.add(tf.layers.conv2d({