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

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.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())
 
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())