Update audio_style_transfer/models/timedomain.py

audio_style_transfer/models/timedomain.py

@@ -244,7 +244,9 @@ def compute_stylization(kernels,
             with tf.compat.v1.Session() as sess:
                 sess.run(tf.compat.v1.initialize_all_variables())
                 print('Started optimization.')
-                opt.minimize(sess)
+                for i in range(iterations):
+                    s, c, l, _ = sess.run([style_loss, content_loss, loss, opt])
+                    print('Style:', s, 'Content:', c, end='\r')
                 result = x.eval()
         else:
             opt = tf.compat.v1.train.AdamOptimizer(