From 3ade542ba526cc27a935fead424ee24c57dd6bb9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Aur=C3=A9lien=20Geron?= Date: Fri, 23 Sep 2022 21:54:25 +1200 Subject: [PATCH] Use verbose=0 when using model.predict() many times --- 15_processing_sequences_using_rnns_and_cnns.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/15_processing_sequences_using_rnns_and_cnns.ipynb b/15_processing_sequences_using_rnns_and_cnns.ipynb index d836ab9..91861c2 100644 --- a/15_processing_sequences_using_rnns_and_cnns.ipynb +++ b/15_processing_sequences_using_rnns_and_cnns.ipynb @@ -4668,7 +4668,7 @@ " arpegio = tf.reshape(arpegio, [1, -1])\n", " for chord in range(length):\n", " for note in range(4):\n", - " next_note = model.predict(arpegio).argmax(axis=-1)[:1, -1:]\n", + " next_note = model.predict(arpegio, verbose=0).argmax(axis=-1)[:1, -1:]\n", " arpegio = tf.concat([arpegio, next_note], axis=1)\n", " arpegio = tf.where(arpegio == 0, arpegio, arpegio + min_note - 1)\n", " return tf.reshape(arpegio, shape=[-1, 4])"