From aeca83bc85adabdefa0e75db470379b878e54051 Mon Sep 17 00:00:00 2001 From: Hao Date: Sun, 28 Jul 2019 22:42:58 +0800 Subject: [PATCH] Update train.py --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 41db4b40..66ebf38f 100755 --- a/train.py +++ b/train.py @@ -73,7 +73,7 @@ def _map_fn_train(img): def train(): G = get_G((batch_size, 96, 96, 3)) D = get_D((batch_size, 384, 384, 3)) - VGG = tl.models.vgg19(pretrained=False, end_with='pool4', mode='static') + VGG = tl.models.vgg19(pretrained=True, end_with='pool4', mode='static') lr_v = tf.Variable(lr_init) g_optimizer_init = tf.optimizers.Adam(lr_v, beta_1=beta1)