Fix Hydra config error in dev script

- Change train.n_iters to train.n_epochs (correct DPPO parameter)
- Update experiment tracking with failed job details
- Ready for corrected dev test
This commit is contained in:
ys1087@partner.kit.edu
2025-08-27 12:07:38 +02:00
parent f88a5be4fe
commit 4adf67694a
2 changed files with 9 additions and 6 deletions
+3 -4
View File
@@ -41,12 +41,11 @@ echo "PyTorch version: $(python -c 'import torch; print(torch.__version__)')"
echo "CUDA available: $(python -c 'import torch; print(torch.cuda.is_available())')"
echo ""
# Run a quick pre-training test with reduced steps
# Run a quick pre-training test with reduced epochs
python script/run.py --config-name=pre_diffusion_mlp \
--config-dir=cfg/gym/pretrain/hopper-medium-v2 \
train.n_iters=10 \
train.log_interval=5 \
train.checkpoint_interval=10 \
train.n_epochs=2 \
train.save_model_freq=1 \
wandb=${WANDB_MODE:-null}
echo "Dev test completed!"