diff --git a/configs/iccv/config_abl_pyr_32_64_128_seq.py b/configs/iccv/config_abl_pyr_32_64_128_seq.py
index 8605270..ece75a1 100644
--- a/configs/iccv/config_abl_pyr_32_64_128_seq.py
+++ b/configs/iccv/config_abl_pyr_32_64_128_seq.py
@@ -1,6 +1,6 @@
 config = dict()
 
-config['gpu'] = '4'
+config['gpu'] = '1'
 config['name'] = 'iccv_pyr_32_64_128_seq_2'
 config['notes'] = ''
 config['tags'] = ['pyr', 'iccv']
@@ -32,7 +32,7 @@ config['pyramid']['final_upscale_enhancer'] = ''
 
 config['train_dir'] = '/home/workspace/local-data/train'
 config['valid_dir'] = '/home/workspace/local-data/valid'
-config['test_dir'] = '/home/workspace/local-data_127/test_ready'
+config['test_dir'] = '/home/workspace/local-data/test_ready'
 
 config['batch_size'] = 4
 config['is_skirt'] = False
diff --git a/configs/iccv/config_abl_pyr_32_seq.py b/configs/iccv/config_abl_pyr_32_seq.py
index f8627bb..8ef53d7 100644
--- a/configs/iccv/config_abl_pyr_32_seq.py
+++ b/configs/iccv/config_abl_pyr_32_seq.py
@@ -32,7 +32,7 @@ config['pyramid']['final_upscale_enhancer'] = ''
 
 config['train_dir'] = '/home/workspace/local-data/train'
 config['valid_dir'] = '/home/workspace/local-data/valid'
-config['test_dir'] = '/home/workspace/local-data_127/test_ready'
+config['test_dir'] = '/home/workspace/local-data/test_ready'
 
 config['batch_size'] = 4
 config['is_skirt'] = False
diff --git a/evaluate_ae_pyramid.py b/evaluate_ae_pyramid.py
index 6e965a5..c155a29 100644
--- a/evaluate_ae_pyramid.py
+++ b/evaluate_ae_pyramid.py
@@ -1,6 +1,6 @@
 import os
 import sys
-from configs.iccv.config_abl_pyr_32_64_128_256_seq import config
+from configs.iccv.config_abl_pyr_32_64_128_seq import config
 
 os.environ["CUDA_VISIBLE_DEVICES"] = config['gpu']
 import tensorflow as tf