From c49908c0ed743766c0604e61c6801bb39a0efcbf Mon Sep 17 00:00:00 2001
From: berak <px1704@web.de>
Date: Sun, 5 Jan 2020 15:54:31 +0100
Subject: [PATCH] Update config.py

so gan_test.py can run out-of-the-box

nice work ;) !
---
 config.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/config.py b/config.py
index 82eea53..5dad7ea 100644
--- a/config.py
+++ b/config.py
@@ -28,7 +28,7 @@
     'collate_fn': get_collate_fn,
 
     # test data
-    'test_data': 'data/test_data/'
+    'test_data': 'data/example_data/'
 }
 
 
@@ -47,7 +47,7 @@
         'input_channel': 64,
         'channels': [1024, 512, 256, 128, 64], # channels for deconvolutional layers
         'output_channel': 3, # images with RGB channels
-        'model_path': 'models/generator.pth',
+        'model_path': 'pretrained_models/generator.pth',
     },
     # FACE EMBEDDING NETWORK (f)
     'f': {