diff --git a/examples/behavior_1k/aggregate.slurm b/examples/behavior_1k/aggregate.slurm index 241a09a9d..5f69edc2e 100644 --- a/examples/behavior_1k/aggregate.slurm +++ b/examples/behavior_1k/aggregate.slurm @@ -4,10 +4,10 @@ #SBATCH --qos=high #SBATCH -c 2 #SBATCH -t 20:00:00 -#SBATCH --mem=8G +#SBATCH --mem=4G #SBATCH -D /admin/home/francesco_capuano/lerobot -#SBATCH -o /admin/home/francesco_capuano/lerobot/examples/behavior_1k/logs/aggregate-50-%j.out -#SBATCH -e /admin/home/francesco_capuano/lerobot/examples/behavior_1k/logs/aggregate-50-%j.err +#SBATCH -o /admin/home/francesco_capuano/lerobot/examples/behavior_1k/logs/%x-%j.out +#SBATCH -e /admin/home/francesco_capuano/lerobot/examples/behavior_1k/logs/%x-%j.err set -euo pipefail set -x @@ -21,4 +21,9 @@ if ! command -v conda >/dev/null 2>&1; then fi conda activate lerobot -python examples/behavior_1k/aggregate_tasks_datasets.py +python examples/behavior_1k/aggregate_tasks_datasets.py \ + --task-datasets-dir /fsx/francesco_capuano/behavior1k-v3 \ + --aggregated-root /fsx/francesco_capuano/behavior1k-v3/behavior1k \ + --num-tasks 50 \ + --hf-user fracapuano \ + --push-to-hub