From a95116f71e2181b0ae36269c21d4ae696d3c6128 Mon Sep 17 00:00:00 2001 From: TheConfused1 <36102900+TheConfused1@users.noreply.github.com> Date: Wed, 31 Oct 2018 21:43:19 +0530 Subject: [PATCH] Update Training.py --- Training.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)