From b438e48c266ee3b80a96395bd92787cef4426df5 Mon Sep 17 00:00:00 2001 From: TheConfused1 <36102900+TheConfused1@users.noreply.github.com> Date: Wed, 31 Oct 2018 21:32:59 +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..e887c6a 100644 --- a/Training.py +++ b/Training.py @@ -4,7 +4,7 @@ from Variables import CONFIG import numpy as np -def content_cost(a_C, a_G): +def content_cost(a_C, a_G): #computes content cost m, n_H, n_W, n_C = a_G.get_shape().as_list() a_C_unrolled = tf.transpose(tf.reshape(a_C,shape=(m,n_H*n_W,n_C))) a_G_unrolled = tf.transpose(tf.reshape(a_G,shape=(m,n_H*n_W,n_C)))