save

configs/iccv/config_abl_all_cape_f.py

@@ -18,8 +18,8 @@ config['use_norm_loss'] =             True
 ablation_count = [config['use_skinning'], config['remove_global_orientation'], config['use_curvature'], config['use_extra_losses'], config['use_offsets']].count(False)
 assert ablation_count == 1 or ablation_count == 0
 
-config['gpu'] = '6'
-config['name'] = 'abl_all_cape_f'
+config['gpu'] = '2'
+config['name'] = 'abl_all_cape_f_2'
 config['notes'] = ''
 config['tags'] = ['iccv', 'abl_all']
 
@@ -50,7 +50,7 @@ if config['is_cape']:
 
 config['latent_dim'] = 128
 
-config['pretrained_model'] = ''
+config['pretrained_model'] = 'checkpoints/iccv/abl_all_cape_f'
 
 ###############    
 config['lr'] = 0.0001#                 0.0001