luni/cvpr
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

also use vae latent for second level

Browse Source
This commit is contained in:
luni
2023-01-30 19:51:14 +00:00
parent 2b8abd776b
commit 6e24c1f41b
5 changed files with 21 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
cvae_pyramid.py
+2 -3
View File
@@ -807,9 +807,8 @@ if __name__ == '__main__':
# assert len(config['pyramid']['resolutions']) == 1
load_model(model, config['pretrained_model'], iteration=config['pretrained_model_iteration'])
model.static_enc.trainable = False
model.body_enc.trainable = False
model.curv_enc.trainable = False
model.trainable = False
model.cond_encoder_2.trainable = True
# if len(config['pyramid']['resolutions']) == 1:
# else:
# load_model(model, config['pyramid']['pretrained_pyramid'], iteration=(config['pyramid']['pretrained_pyramid_iteration'] if 'pretrained_pyramid_iteration' in config.keys() else ''))