diff --git a/Training.py b/Training.py index 8b36c34..a06a341 100644 --- a/Training.py +++ b/Training.py @@ -53,7 +53,7 @@ def total_cost(J_content, J_style, alpha = 10, beta = 40): sess.run(model['input'].assign(style_image)) J_style = compute_style_cost(model, CONFIG.STYLE_LAYERS) -J = total_cost(J_content, J_style,alpha=CONFIG.ALPHA,beta=CONFIG.BETA) +J = total_cost(J_content, J_style,alpha=CONFIG.ALPHA,beta=CONFIG.BETA) #comprises of both style and content losses optimizer = tf.train.AdamOptimizer(CONFIG.LEARNINF_RATE) train_step = optimizer.minimize(J)