Upload paper_dsprites_cnn extrapolation seed 1
Browse files
checkpoints/paper_dsprites_cnn/dsprites/extrapolation/cnn/seed_01/last.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:44068ec0000b5bc7651d1c67c776302a117319542faa7e4df96425f2f7697b79
|
| 3 |
+
size 4563526
|
checkpoints/paper_dsprites_cnn/dsprites/extrapolation/cnn/seed_01/resolved_config.yaml
ADDED
|
@@ -0,0 +1,65 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
experiment:
|
| 2 |
+
name: paper_dsprites_cnn
|
| 3 |
+
description: Paper Appendix H.3 supervised CNN baseline.
|
| 4 |
+
matrix:
|
| 5 |
+
variants:
|
| 6 |
+
- random
|
| 7 |
+
- composition
|
| 8 |
+
- interpolation
|
| 9 |
+
- extrapolation
|
| 10 |
+
seeds:
|
| 11 |
+
- 0
|
| 12 |
+
- 1
|
| 13 |
+
- 2
|
| 14 |
+
max_parallel: 4
|
| 15 |
+
data:
|
| 16 |
+
dataset: dsprites
|
| 17 |
+
variant: extrapolation
|
| 18 |
+
root: experiments/data
|
| 19 |
+
batch_size: 64
|
| 20 |
+
num_workers: 4
|
| 21 |
+
training:
|
| 22 |
+
model: cnn
|
| 23 |
+
steps: 500000
|
| 24 |
+
learning_rate: 0.0005
|
| 25 |
+
seed: 1
|
| 26 |
+
device: auto
|
| 27 |
+
log_every: 1000
|
| 28 |
+
checkpoint_every: 25000
|
| 29 |
+
resume: auto
|
| 30 |
+
results:
|
| 31 |
+
root: /net/scratch/mx99/idgb/results
|
| 32 |
+
hub:
|
| 33 |
+
enabled: true
|
| 34 |
+
repo_id: mchlxo/comp-gen-bench
|
| 35 |
+
evaluation:
|
| 36 |
+
enabled: true
|
| 37 |
+
batch_size: 256
|
| 38 |
+
num_workers: 4
|
| 39 |
+
device: auto
|
| 40 |
+
max_parallel: 4
|
| 41 |
+
upload_to_hub: true
|
| 42 |
+
slurm:
|
| 43 |
+
time: 02:00:00
|
| 44 |
+
cpus_per_task: 4
|
| 45 |
+
mem: 24G
|
| 46 |
+
gpus: 1
|
| 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: 10
|
| 60 |
+
size: 12
|
| 61 |
+
variant: extrapolation
|
| 62 |
+
seed: 1
|
| 63 |
+
provenance:
|
| 64 |
+
config_path: /home/mx99/comp-gen-bench/InDomainGeneralizationBenchmark/experiments/configs/paper/dsprites_cnn.yaml
|
| 65 |
+
project_root: /home/mx99/comp-gen-bench/InDomainGeneralizationBenchmark
|
checkpoints/paper_dsprites_cnn/dsprites/extrapolation/cnn/seed_01/status.yaml
ADDED
|
@@ -0,0 +1,14 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
status: completed
|
| 2 |
+
updated_at: '2026-07-26T08:59:03.866037+00:00'
|
| 3 |
+
experiment: paper_dsprites_cnn
|
| 4 |
+
dataset: dsprites
|
| 5 |
+
variant: extrapolation
|
| 6 |
+
model: cnn
|
| 7 |
+
seed: 1
|
| 8 |
+
slurm_job_id: '1212382'
|
| 9 |
+
slurm_array_job_id: '1211524'
|
| 10 |
+
slurm_array_task_id: '10'
|
| 11 |
+
checkpoint: /net/scratch/mx99/idgb/results/paper_dsprites_cnn/dsprites/extrapolation/cnn/seed_01/last.pt
|
| 12 |
+
metrics:
|
| 13 |
+
mse: 0.0017630846705287695
|
| 14 |
+
elapsed_seconds: 3709.560079470277
|