gaojie
c66855adfc
Sync to site1 / sync (push) Has been cancelled
- 移除 JEPA/lejepa-identifiability 子模块 gitlink - 移除 research/multiply/MultiPLY 子模块 gitlink - 删除 .gitmodules（不再有外部 URL 依赖） - 两个目录内容作为普通文件纳入主仓库追踪 - 删除各自内部 .git 目录，消除嵌套 git 仓库
29 lines
648 B
YAML
29 lines
648 B
YAML
# Reacher pixel-observation identifiability experiment
|
|
# Works for both OU and trajectory data — just point --data_dir at the right place.
|
|
#
|
|
# Prerender:
|
|
# python prerender.py eval
|
|
# python prerender.py ou --rho 0.95
|
|
# python prerender.py traj --delta 16 --h5_path data/reacher.h5
|
|
#
|
|
# Train:
|
|
# python run_reacher.py --config configs/reacher.yaml \
|
|
# --data_dir data/reacher/ou/rho=0.95
|
|
|
|
experiment: reacher
|
|
out: results/reacher
|
|
data_root: data/reacher
|
|
|
|
# Model
|
|
d_latent: 2
|
|
|
|
# Training
|
|
epochs: 100
|
|
batch_size: 256
|
|
lr: 3.0e-3
|
|
n_slices: 256
|
|
n_eval_fast: 2000
|
|
|
|
# Sweep dimensions
|
|
lambs: [1.0e-3, 5.0e-3, 1.0e-2, 5.0e-2]
|
|
seeds: [0, 1, 2] |