05dddfa10c
- Add installation script for HoReKa with Python 3.10 venv - Add SLURM job submission scripts for dev and production runs - Add convenient submit_job.sh wrapper for easy job submission - Update .gitignore to allow shell scripts (removed *.sh exclusion) - Configure git remotes: upstream (original) and origin (fork)
46 lines
1.2 KiB
Bash
Executable File
46 lines
1.2 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
# Submit DPPO jobs to SLURM
|
|
# Usage: ./submit_job.sh [mode] [task] [experiment_type]
|
|
|
|
set -e
|
|
|
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
|
cd "$SCRIPT_DIR"
|
|
|
|
# Default values
|
|
MODE=${1:-dev} # dev, gym, robomimic, d3il, furniture
|
|
TASK=${2:-hopper} # Task specific to each mode
|
|
EXPERIMENT_TYPE=${3:-pretrain} # pretrain or finetune
|
|
|
|
echo "Submitting DPPO job..."
|
|
echo "Mode: $MODE"
|
|
echo "Task: $TASK"
|
|
echo "Experiment type: $EXPERIMENT_TYPE"
|
|
|
|
case $MODE in
|
|
dev)
|
|
echo "Submitting development test job..."
|
|
sbatch slurm/run_dppo_dev.sh
|
|
;;
|
|
gym)
|
|
echo "Submitting Gym job..."
|
|
sbatch slurm/run_dppo_gym.sh "$TASK" "$EXPERIMENT_TYPE"
|
|
;;
|
|
*)
|
|
echo "Unknown mode: $MODE"
|
|
echo "Supported modes: dev, gym"
|
|
echo ""
|
|
echo "Usage examples:"
|
|
echo " ./submit_job.sh dev # Run quick dev test"
|
|
echo " ./submit_job.sh gym hopper pretrain # Pre-train on Gym Hopper"
|
|
echo " ./submit_job.sh gym walker2d finetune # Fine-tune on Gym Walker2D"
|
|
exit 1
|
|
;;
|
|
esac
|
|
|
|
echo ""
|
|
echo "Job submitted! Check status with:"
|
|
echo " squeue -u $USER"
|
|
echo ""
|
|
echo "Check logs in: logs/ directory" |