gaojie
c66855adfc
Sync to site1 / sync (push) Has been cancelled
- 移除 JEPA/lejepa-identifiability 子模块 gitlink - 移除 research/multiply/MultiPLY 子模块 gitlink - 删除 .gitmodules（不再有外部 URL 依赖） - 两个目录内容作为普通文件纳入主仓库追踪 - 删除各自内部 .git 目录，消除嵌套 git 仓库
30 lines
1.1 KiB
YAML
30 lines
1.1 KiB
YAML
# 2D illustration + Gaussian half of the regularizer ablation
|
|
# python run.py --config configs/2d.yaml --run spiral_lejepa --seed 1337
|
|
|
|
experiment: 2d
|
|
out: results/2d
|
|
|
|
# Data
|
|
N: 2
|
|
source_dist: gaussian
|
|
num_eval: 10000
|
|
|
|
# Training (shared)
|
|
steps: 20000
|
|
lr: 3.0e-3
|
|
batch_size: 256
|
|
rho: 0.95
|
|
log_every: 500
|
|
|
|
# Per-run specs: 4 mixings x 2 objectives
|
|
runs:
|
|
spiral_lejepa: {mixing: spiral, encoder: mlp, hidden: 256, mode: lejepa, lamb: 1.0e-3}
|
|
spiral_whiten: {mixing: spiral, encoder: mlp, hidden: 256, mode: whiten, lamb: 0.5}
|
|
banana_lejepa: {mixing: banana, encoder: mlp, hidden: 256, mode: lejepa, lamb: 1.0e-3}
|
|
banana_whiten: {mixing: banana, encoder: mlp, hidden: 256, mode: whiten, lamb: 0.5}
|
|
sinusoid_lejepa: {mixing: sinusoid, encoder: mlp, hidden: 256, mode: lejepa, lamb: 1.0e-3}
|
|
sinusoid_whiten: {mixing: sinusoid, encoder: mlp, hidden: 256, mode: whiten, lamb: 0.5}
|
|
nvp_lejepa: {mixing: nvp, encoder: matched, n_layers: 8, mode: lejepa, lamb: 1.0e-3}
|
|
nvp_whiten: {mixing: nvp, encoder: matched, n_layers: 8, mode: whiten, lamb: 0.5}
|
|
|
|
seeds: [1337, 1338, 1339] |