more training

This commit is contained in:
2023-03-02 14:59:51 +00:00
parent 424f8d2fcc
commit 25112a7ff9
17 changed files with 1484 additions and 17 deletions
+3 -3
View File
@@ -18,7 +18,7 @@ config['use_norm_loss'] = False
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'] = '7'
config['gpu'] = '2'
config['name'] = 'abl_skin_reg_skirt'
config['notes'] = ''
config['tags'] = ['iccv', 'abl_skin_reg', 'skirt']
@@ -36,7 +36,7 @@ config['dataset_in_memory'] = False
config['is_cape'] = False
config['train_dir'] = '/home/workspace/local-data_128/train_skirt'
config['valid_dir'] = '/home/workspace/local-data_127/valid_skirt'
config['valid_dir'] = '/home/workspace/local-data/valid_skirt'
config['test_dir'] = '/home/workspace/data/dataset/test_skirt_ready_2'
config['static_model'] = 'checkpoints/iccv/static_iccv_1_skirt'
@@ -50,7 +50,7 @@ if config['is_cape']:
config['latent_dim'] = 128
config['pretrained_model'] = ''
config['pretrained_model'] = 'checkpoints/iccv/abl_skin_reg_skirt'
###############
config['lr'] = 0.0001# 0.0001