Update train.py

tensorlayer · Jul 28, 2019 · aeca83b · aeca83b
1 parent 487aabf
commit aeca83b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)