save

cvae_pyramid.py

@@ -1,6 +1,6 @@
 import os
 import sys
-from configs.iccv.config_abl_pyr_32_64_128_reg import config
+from configs.iccv.config_abl_pyr_32_64_128_256_512_curv import config
 
 os.environ["CUDA_VISIBLE_DEVICES"] = config['gpu']
 import tensorflow as tf
@@ -826,7 +826,7 @@ if __name__ == '__main__':
     # model.levels[0].upscale_enhancer.trainable = False
 
 
-    for i in range(len(model.levels)):
+    for i in range(len(model.levels)-1):
         load_model(model.levels[i].upscale_enhancer, config['pyramid']['levels'][i]['upscale_enhancer'])
         model.levels[i].upscale_enhancer.trainable = False
     # if config['pyramid']['resolutions'][-1] != 512: