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

Improve logging

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';
export default function(cats_count, image_size, use_crossentropy = false, activation = "relu") {
console.error(`>>> Creating new model (activation = ${activation})`);
let model = tf.sequential();
model.add(tf.layers.conv2d({