diff --git a/model-classification-app/scripts/transfer-learning.js b/model-classification-app/scripts/transfer-learning.js index 4b680d4..dba6bec 100644 --- a/model-classification-app/scripts/transfer-learning.js +++ b/model-classification-app/scripts/transfer-learning.js @@ -57,6 +57,8 @@ async function run() { }); await classifyAllImages(mobileNetModel, myTransferMobileNetModel); + + console.log('Classification complete!'); } /* Functional implementation */