also use vae latent for second level

2023-01-30 19:51:14 +00:00
parent 2b8abd776b
commit 6e24c1f41b
5 changed files with 21 additions and 9 deletions
@@ -1,7 +1,7 @@
 config = dict()

 config['gpu'] = '0'
-config['name'] = 'pyr_abl_32_64_e2e_r3'
+config['name'] = 'pyr_abl_32_64_e2e_r3_pre'
 config['notes'] = ''
 config['tags'] = ['review_3']

@@ -26,7 +26,7 @@ config['pyramid']['resolutions'] = [32, 64]
 config['pyramid']['pretrained_pyramid'] = ''
 config['pyramid']['final_upscale_enhancer'] = 'checkpoints/64_512_2'

-config['pretrained_model'] = ''
+config['pretrained_model'] = 'checkpoints/pyr_abl_32_64_e2e_r3'
 config['pretrained_model_iteration'] = ''

 config['train_dir'] = '/home/workspace/local-data/train'