nakas commited on
Commit
27110eb
·
1 Parent(s): c3fb333

Update audio_style_transfer/models/timedomain.py

Browse files
audio_style_transfer/models/timedomain.py CHANGED
@@ -238,8 +238,8 @@ def compute_stylization(kernels,
238
  layer_i + 1])
239
  loss = content_loss + style_loss
240
  if optimizer == 'bfgs':
241
- opt = tf.contrib.opt.ScipyOptimizerInterface(
242
- loss, method='L-BFGS-B', options={'maxiter': iterations})
243
  # Optimization
244
  with tf.compat.v1.Session() as sess:
245
  sess.run(tf.compat.v1.initialize_all_variables())
 
238
  layer_i + 1])
239
  loss = content_loss + style_loss
240
  if optimizer == 'bfgs':
241
+ opt = tf.compat.v1.train.AdamOptimizer(
242
+ learning_rate=learning_rate).minimize(loss)
243
  # Optimization
244
  with tf.compat.v1.Session() as sess:
245
  sess.run(tf.compat.v1.initialize_all_variables())