fix: minor things on the aggregation job

This commit is contained in:
fracapuano
2025-11-21 09:29:07 +00:00
parent 2b2ff19366
commit 1c8f922379
+9 -4
View File
@@ -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