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

fine tune

Browse Source
This commit is contained in:
luni
2023-01-31 16:31:30 +00:00
parent 6e24c1f41b
commit 2489e28de4
3 changed files with 11 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
cvae_pyramid.py
+6 -2
View File
@@ -807,8 +807,12 @@ if __name__ == '__main__':
# assert len(config['pyramid']['resolutions']) == 1
load_model(model, config['pretrained_model'], iteration=config['pretrained_model_iteration'])
model.trainable = False
model.cond_encoder_2.trainable = True
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 ''))