Spaces:
Runtime error
Runtime error
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 |
-
|
| 242 |
-
|
| 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())
|