diff --git a/src/lib/model.mjs b/src/lib/model.mjs index b20e143..b4a960d 100644 --- a/src/lib/model.mjs +++ b/src/lib/model.mjs @@ -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({