Hugging Face's logo

mchlxo
/
comp-gen-bench

PyTorch
domain-generalization
disentanglement
dsprites
Model card Files
xet
 Community
mchlxo commited on
Commit
69b04ee
·
verified ·
1 Parent(s): 00298af

Upload paper_dsprites_beta_vae interpolation seed 6

Browse files
Files changed (3) hide show
  1. checkpoints/paper_dsprites_beta_vae/dsprites/interpolation/beta_vae/seed_06/last.pt +3 -0
  2. checkpoints/paper_dsprites_beta_vae/dsprites/interpolation/beta_vae/seed_06/resolved_config.yaml +65 -0
  3. checkpoints/paper_dsprites_beta_vae/dsprites/interpolation/beta_vae/seed_06/status.yaml +16 -0
checkpoints/paper_dsprites_beta_vae/dsprites/interpolation/beta_vae/seed_06/last.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1cbb7d7c238b74bd58dad588856265d251e9fdccc24faa0fe41ed89377e89eb6
3
+ size 8317410
checkpoints/paper_dsprites_beta_vae/dsprites/interpolation/beta_vae/seed_06/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: interpolation
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: 6
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: 26
60
+ size: 40
61
+ variant: interpolation
62
+ seed: 6
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/interpolation/beta_vae/seed_06/status.yaml ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ status: completed
2
+ updated_at: '2026-07-26T11:38:07.433109+00:00'
3
+ experiment: paper_dsprites_beta_vae
4
+ dataset: dsprites
5
+ variant: interpolation
6
+ model: beta_vae
7
+ seed: 6
8
+ slurm_job_id: '1214668'
9
+ slurm_array_job_id: '1211523'
10
+ slurm_array_task_id: '26'
11
+ checkpoint: /net/scratch/mx99/idgb/results/paper_dsprites_beta_vae/dsprites/interpolation/beta_vae/seed_06/last.pt
12
+ metrics:
13
+ loss: 44.436309814453125
14
+ reconstruction_loss: 16.628494262695312
15
+ kl_loss: 27.807815551757812
16
+ elapsed_seconds: 1903.1184594929218