use jumpsuit overfit model

This commit is contained in:
2023-01-28 19:57:23 +00:00
parent 54a77273c5
commit cc723222dc
2 changed files with 4 additions and 4 deletions
+2 -2
View File
@@ -19,8 +19,8 @@ config['only_curvatures'] = 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'] = '5'
config['name'] = 'abl_overfit_jumpsuit_2'
config['gpu'] = '3'
config['name'] = 'abl_overfit_jumpsuit_2_r4'
config['notes'] = 'remove other conditions from input as well'
config['avg_garment_path'] = 'custom_data/train_jumpsuit_30k_no_shape_avg_uv.npy' #custom_data/train_no_shape_avg_uv.npy # train_no_shape_avg_uv train_no_shape_one_hot_avg_uv