Hugging Face's logo

Inbar2344
/
AnyTop

Model card Files
xet
 Community
Inbar2344 commited on
Commit
74c8bcc
·
verified ·
1 Parent(s): c994004

Upload 15 files

Browse files
Files changed (15) hide show
  1. checkpoints/.DS_Store +0 -0
  2. checkpoints/all_model_dataset_truebones_bs_16_latentdim_128/.DS_Store +0 -0
  3. checkpoints/all_model_dataset_truebones_bs_16_latentdim_128/args.json +44 -0
  4. checkpoints/all_model_dataset_truebones_bs_16_latentdim_128/model000369999.pt +3 -0
  5. checkpoints/all_model_dataset_truebones_bs_16_latentdim_128/opt000370000.pt +3 -0
  6. checkpoints/bipeds_model_dataset_truebones_bs_16_latentdim_128/args.json +44 -0
  7. checkpoints/bipeds_model_dataset_truebones_bs_16_latentdim_128/model000359999.pt +3 -0
  8. checkpoints/bipeds_model_dataset_truebones_bs_16_latentdim_128/opt000360000.pt +3 -0
  9. checkpoints/flying_model_dataset_truebones_bs_16_latentdim_128/args.json +44 -0
  10. checkpoints/flying_model_dataset_truebones_bs_16_latentdim_128/model000279999.pt +3 -0
  11. checkpoints/flying_model_dataset_truebones_bs_16_latentdim_128/opt000290000.pt +3 -0
  12. checkpoints/millipeds_snakes_model_dataset_truebones_bs_16_latentdim_128/args.json +44 -0
  13. checkpoints/millipeds_snakes_model_dataset_truebones_bs_16_latentdim_128/model000379999.pt +3 -0
  14. checkpoints/millipeds_snakes_model_dataset_truebones_bs_16_latentdim_128/opt000380000.pt +3 -0
  15. checkpoints/quadropeds_model_dataset_truebones_bs_16_latentdim_128/model000399999.pt +3 -0
checkpoints/.DS_Store ADDED
Binary file (6.15 kB). View file
 
checkpoints/all_model_dataset_truebones_bs_16_latentdim_128/.DS_Store ADDED
Binary file (6.15 kB). View file
 
checkpoints/all_model_dataset_truebones_bs_16_latentdim_128/args.json ADDED
@@ -0,0 +1,44 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "arch": "trans_enc",
3
+ "balanced": true,
4
+ "batch_size": 16,
5
+ "cond_mask_prob": 0.1,
6
+ "cuda": true,
7
+ "data_dir": "",
8
+ "device": 0,
9
+ "diffusion_steps": 100,
10
+ "emb_trans_dec": false,
11
+ "eval_batch_size": 32,
12
+ "eval_during_training": false,
13
+ "eval_num_samples": 1000,
14
+ "eval_rep_times": 3,
15
+ "eval_split": "test",
16
+ "gen_during_training": true,
17
+ "gen_num_repetitions": 2,
18
+ "gen_num_samples": 3,
19
+ "lambda_fs": 0.0,
20
+ "lambda_geo": 1.0,
21
+ "latent_dim": 128,
22
+ "layers": 4,
23
+ "log_interval": 50,
24
+ "lr": 0.0001,
25
+ "lr_anneal_steps": 0,
26
+ "model_prefix": "all_model",
27
+ "noise_schedule": "cosine",
28
+ "num_frames": 120,
29
+ "num_steps": 600000,
30
+ "objects_subset": "all",
31
+ "overwrite": true,
32
+ "resume_checkpoint": "",
33
+ "save_dir": "/home/dcor/inbargat1/multi-skeleton-mdm/save/all_model_dataset_truebones_bs_16_latentdim_128",
34
+ "save_interval": 10000,
35
+ "seed": 10,
36
+ "sigma_small": true,
37
+ "skip_t5": false,
38
+ "t5_name": "t5-base",
39
+ "temporal_window": 31,
40
+ "train_platform_type": "WandBPlatform",
41
+ "use_ema": false,
42
+ "value_emb": false,
43
+ "weight_decay": 0.0
44
+ }
checkpoints/all_model_dataset_truebones_bs_16_latentdim_128/model000369999.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d26c8eb0e545bc327ac82dbd139b0b18e863fabbbd96ee3bcd9ab882c8951e16
3
+ size 9223630
checkpoints/all_model_dataset_truebones_bs_16_latentdim_128/opt000370000.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dee1a7391670143f3eaf1539d5b6fb7d56f736c3da88bfa622146d21a55f4203
3
+ size 14198982
checkpoints/bipeds_model_dataset_truebones_bs_16_latentdim_128/args.json ADDED
@@ -0,0 +1,44 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "arch": "trans_enc",
3
+ "balanced": true,
4
+ "batch_size": 16,
5
+ "cond_mask_prob": 0.1,
6
+ "cuda": true,
7
+ "data_dir": "",
8
+ "device": 0,
9
+ "diffusion_steps": 100,
10
+ "emb_trans_dec": false,
11
+ "eval_batch_size": 32,
12
+ "eval_during_training": false,
13
+ "eval_num_samples": 1000,
14
+ "eval_rep_times": 3,
15
+ "eval_split": "test",
16
+ "gen_during_training": true,
17
+ "gen_num_repetitions": 2,
18
+ "gen_num_samples": 3,
19
+ "lambda_fs": 0.0,
20
+ "lambda_geo": 1.0,
21
+ "latent_dim": 128,
22
+ "layers": 4,
23
+ "log_interval": 50,
24
+ "lr": 0.0001,
25
+ "lr_anneal_steps": 0,
26
+ "model_prefix": "bipeds_model",
27
+ "noise_schedule": "cosine",
28
+ "num_frames": 120,
29
+ "num_steps": 600000,
30
+ "objects_subset": "bipeds",
31
+ "overwrite": true,
32
+ "resume_checkpoint": "",
33
+ "save_dir": "/home/dcor/inbargat1/multi-skeleton-mdm/save/bipeds_model_dataset_truebones_bs_16_latentdim_128",
34
+ "save_interval": 10000,
35
+ "seed": 10,
36
+ "sigma_small": true,
37
+ "skip_t5": false,
38
+ "t5_name": "t5-base",
39
+ "temporal_window": 31,
40
+ "train_platform_type": "WandBPlatform",
41
+ "use_ema": false,
42
+ "value_emb": false,
43
+ "weight_decay": 0.0
44
+ }
checkpoints/bipeds_model_dataset_truebones_bs_16_latentdim_128/model000359999.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2dca8d96c555b68597547116169084202d336d90a75bd9b77a00ccbee1c1c511
3
+ size 9223630
checkpoints/bipeds_model_dataset_truebones_bs_16_latentdim_128/opt000360000.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3bc208450e37dd4c7dca6fdaac7b18c9f37e9c3b979af317f95ab6e76b361254
3
+ size 14198982
checkpoints/flying_model_dataset_truebones_bs_16_latentdim_128/args.json ADDED
@@ -0,0 +1,44 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "arch": "trans_enc",
3
+ "balanced": true,
4
+ "batch_size": 16,
5
+ "cond_mask_prob": 0.1,
6
+ "cuda": true,
7
+ "data_dir": "",
8
+ "device": 0,
9
+ "diffusion_steps": 100,
10
+ "emb_trans_dec": false,
11
+ "eval_batch_size": 32,
12
+ "eval_during_training": false,
13
+ "eval_num_samples": 1000,
14
+ "eval_rep_times": 3,
15
+ "eval_split": "test",
16
+ "gen_during_training": true,
17
+ "gen_num_repetitions": 2,
18
+ "gen_num_samples": 3,
19
+ "lambda_fs": 0.0,
20
+ "lambda_geo": 1.0,
21
+ "latent_dim": 128,
22
+ "layers": 4,
23
+ "log_interval": 50,
24
+ "lr": 0.0001,
25
+ "lr_anneal_steps": 0,
26
+ "model_prefix": "flying_model",
27
+ "noise_schedule": "cosine",
28
+ "num_frames": 120,
29
+ "num_steps": 600000,
30
+ "objects_subset": "flying",
31
+ "overwrite": true,
32
+ "resume_checkpoint": "",
33
+ "save_dir": "/home/dcor/inbargat1/multi-skeleton-mdm/save/flying_model_dataset_truebones_bs_16_latentdim_128",
34
+ "save_interval": 10000,
35
+ "seed": 10,
36
+ "sigma_small": true,
37
+ "skip_t5": false,
38
+ "t5_name": "t5-base",
39
+ "temporal_window": 31,
40
+ "train_platform_type": "WandBPlatform",
41
+ "use_ema": false,
42
+ "value_emb": false,
43
+ "weight_decay": 0.0
44
+ }
checkpoints/flying_model_dataset_truebones_bs_16_latentdim_128/model000279999.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d8c1bc600eec8f0b17adaffe587383abbd23b2027c0d47c6254f28218288abc3
3
+ size 9223630
checkpoints/flying_model_dataset_truebones_bs_16_latentdim_128/opt000290000.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ca07b3f2275bb416dea3886071786fa4f557f4ffebe69d22c9c230b3af21f235
3
+ size 14198982
checkpoints/millipeds_snakes_model_dataset_truebones_bs_16_latentdim_128/args.json ADDED
@@ -0,0 +1,44 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "arch": "trans_enc",
3
+ "balanced": true,
4
+ "batch_size": 16,
5
+ "cond_mask_prob": 0.1,
6
+ "cuda": true,
7
+ "data_dir": "",
8
+ "device": 0,
9
+ "diffusion_steps": 100,
10
+ "emb_trans_dec": false,
11
+ "eval_batch_size": 32,
12
+ "eval_during_training": false,
13
+ "eval_num_samples": 1000,
14
+ "eval_rep_times": 3,
15
+ "eval_split": "test",
16
+ "gen_during_training": true,
17
+ "gen_num_repetitions": 2,
18
+ "gen_num_samples": 3,
19
+ "lambda_fs": 0.0,
20
+ "lambda_geo": 1.0,
21
+ "latent_dim": 128,
22
+ "layers": 4,
23
+ "log_interval": 50,
24
+ "lr": 0.0001,
25
+ "lr_anneal_steps": 0,
26
+ "model_prefix": "millipeds_snakes_model",
27
+ "noise_schedule": "cosine",
28
+ "num_frames": 120,
29
+ "num_steps": 600000,
30
+ "objects_subset": "millipeds_snakes",
31
+ "overwrite": true,
32
+ "resume_checkpoint": "",
33
+ "save_dir": "/home/dcor/inbargat1/multi-skeleton-mdm/save/millipeds_snakes_model_dataset_truebones_bs_16_latentdim_128",
34
+ "save_interval": 10000,
35
+ "seed": 10,
36
+ "sigma_small": true,
37
+ "skip_t5": false,
38
+ "t5_name": "t5-base",
39
+ "temporal_window": 31,
40
+ "train_platform_type": "WandBPlatform",
41
+ "use_ema": false,
42
+ "value_emb": false,
43
+ "weight_decay": 0.0
44
+ }
checkpoints/millipeds_snakes_model_dataset_truebones_bs_16_latentdim_128/model000379999.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d3318b0f8fc6326113548e7801f3b1232bfe8a687581a80629a067b61ca4ec36
3
+ size 9223630
checkpoints/millipeds_snakes_model_dataset_truebones_bs_16_latentdim_128/opt000380000.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6207594eb611ee05b1a0daa580ea9796938a6fe6534fe60f0c012bc2f1e46942
3
+ size 14198982
checkpoints/quadropeds_model_dataset_truebones_bs_16_latentdim_128/model000399999.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e8e2a96745f2e619b8ee8ee6af751156aa4ec1bff04781edd875bb8f8ab4409d
3
+ size 9223630