Hugging Face's logo

mchlxo
/
comp-gen-bench

PyTorch
domain-generalization
disentanglement
dsprites
Model card Files
xet
 Community
mchlxo commited on
Commit
846c459
·
verified ·
1 Parent(s): 5fd5db4

Upload paper_dsprites_beta_vae extrapolation seed 1

Browse files
Files changed (3) hide show
  1. checkpoints/paper_dsprites_beta_vae/dsprites/extrapolation/beta_vae/seed_01/last.pt +3 -0
  2. checkpoints/paper_dsprites_beta_vae/dsprites/extrapolation/beta_vae/seed_01/resolved_config.yaml +65 -0
  3. checkpoints/paper_dsprites_beta_vae/dsprites/extrapolation/beta_vae/seed_01/status.yaml +16 -0
checkpoints/paper_dsprites_beta_vae/dsprites/extrapolation/beta_vae/seed_01/last.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:45a2e7e59900c1baa1642d5b368b08ef935fb7ba09796d817417f8c5ca6479c9
3
+ size 8317410
checkpoints/paper_dsprites_beta_vae/dsprites/extrapolation/beta_vae/seed_01/resolved_config.yaml ADDED
@@ -0,0 +1,65 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ experiment:
2
+ name: paper_dsprites_beta_vae
3
+ description: Paper Appendix H.3 beta-VAE representation model.
4
+ matrix:
5
+ variants:
6
+ - random
7
+ - composition
8
+ - interpolation
9
+ - extrapolation
10
+ seeds:
11
+ - 0
12
+ - 1
13
+ - 2
14
+ - 3
15
+ - 4
16
+ - 5
17
+ - 6
18
+ - 7
19
+ - 8
20
+ - 9
21
+ max_parallel: 4
22
+ data:
23
+ dataset: dsprites
24
+ variant: extrapolation
25
+ root: experiments/data
26
+ batch_size: 64
27
+ num_workers: 4
28
+ training:
29
+ model: beta_vae
30
+ steps: 500000
31
+ learning_rate: 0.0001
32
+ beta: 1.0
33
+ latent_dim: 10
34
+ reconstruction: bernoulli
35
+ seed: 1
36
+ device: auto
37
+ log_every: 1000
38
+ checkpoint_every: 25000
39
+ resume: auto
40
+ results:
41
+ root: /net/scratch/mx99/idgb/results
42
+ hub:
43
+ enabled: true
44
+ repo_id: mchlxo/comp-gen-bench
45
+ evaluation:
46
+ enabled: false
47
+ slurm:
48
+ partition: general
49
+ account: general_group
50
+ time: '11:55:00'
51
+ cpus_per_task: 4
52
+ mem: 32G
53
+ gpus: 1
54
+ signal_seconds: 300
55
+ environment: .micromamba/cluster-env
56
+ log_root: experiments/logs
57
+ hf_home: /net/scratch/mx99/idgb/huggingface
58
+ matrix_selection:
59
+ index: 31
60
+ size: 40
61
+ variant: extrapolation
62
+ seed: 1
63
+ provenance:
64
+ config_path: /home/mx99/comp-gen-bench/InDomainGeneralizationBenchmark/experiments/configs/paper/dsprites_beta_vae.yaml
65
+ project_root: /home/mx99/comp-gen-bench/InDomainGeneralizationBenchmark
checkpoints/paper_dsprites_beta_vae/dsprites/extrapolation/beta_vae/seed_01/status.yaml ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ status: completed
2
+ updated_at: '2026-07-26T12:23:15.492640+00:00'
3
+ experiment: paper_dsprites_beta_vae
4
+ dataset: dsprites
5
+ variant: extrapolation
6
+ model: beta_vae
7
+ seed: 1
8
+ slurm_job_id: '1215232'
9
+ slurm_array_job_id: '1211523'
10
+ slurm_array_task_id: '31'
11
+ checkpoint: /net/scratch/mx99/idgb/results/paper_dsprites_beta_vae/dsprites/extrapolation/beta_vae/seed_01/last.pt
12
+ metrics:
13
+ loss: 35.913658142089844
14
+ reconstruction_loss: 12.255929946899414
15
+ kl_loss: 23.65772819519043
16
+ elapsed_seconds: 2206.125842498004