Create task.yaml
This commit is contained in:
parent
f39652707c
commit
8c2070a14b
|
@ -0,0 +1,87 @@
|
||||||
|
# https://taskfile.dev
|
||||||
|
|
||||||
|
version: '3'
|
||||||
|
|
||||||
|
vars:
|
||||||
|
PROJECT_DIR: "{{.ROOT_DIR}}"
|
||||||
|
SRC_DIR: "{{.PROJECT_DIR}/lerobot"
|
||||||
|
TESTS_DIR: "{{.PROJECT_DIR}}/tests"
|
||||||
|
EXAMPLES_DIR: "{{.PROJECT_DIR}}/examples"
|
||||||
|
DATA_DIR: "{{.PROJECT_DIR}}/media"
|
||||||
|
|
||||||
|
tasks:
|
||||||
|
default:
|
||||||
|
desc: Liste toutes les tâches disponibles
|
||||||
|
cmds:
|
||||||
|
- task --list
|
||||||
|
silent: true
|
||||||
|
|
||||||
|
get-prerequisites-mac:
|
||||||
|
desc: Installe les prérequis sur mac
|
||||||
|
cmds:
|
||||||
|
- brew install python@3.10 task
|
||||||
|
- pip install uv pipenv --break-system-packages
|
||||||
|
|
||||||
|
shell:
|
||||||
|
desc: Ouvre un shell python avec les dépendances du projet
|
||||||
|
cmds:
|
||||||
|
- pipenv shell "cd lerobot"
|
||||||
|
|
||||||
|
prepare-instance:
|
||||||
|
desc: Prépare une instance pour le projet
|
||||||
|
cmds:
|
||||||
|
- uv pip install -e ".[aloha, pusht]"
|
||||||
|
|
||||||
|
install:
|
||||||
|
desc: Installe les dépendances du projet robot follow
|
||||||
|
cmds:
|
||||||
|
- uv pip install -e ".[feetech]"
|
||||||
|
|
||||||
|
wandb-login:
|
||||||
|
desc: Connecte à wandb
|
||||||
|
cmds:
|
||||||
|
- wandb login
|
||||||
|
|
||||||
|
run-example:
|
||||||
|
desc: Exécute un exemple
|
||||||
|
cmds:
|
||||||
|
- python lerobot/scripts/visualize_dataset.py --repo-id lerobot/pusht --episode-index 0
|
||||||
|
|
||||||
|
evaluate-policy:
|
||||||
|
test-unit:
|
||||||
|
desc: Evalue une policy pretrainer
|
||||||
|
cmds:
|
||||||
|
- python lerobot/scripts/eval.py --policy.path=lerobot/diffusion_pusht --env.type=pusht --eval.batch_size=10 --eval.n_episodes=10 --policy.use_amp=false --policy.device=cuda
|
||||||
|
|
||||||
|
set-usb-motor-id:
|
||||||
|
desc: Définit l'ID des moteurs
|
||||||
|
args:
|
||||||
|
ID:
|
||||||
|
shorthand: i
|
||||||
|
desc: ID du moteur
|
||||||
|
PORT:
|
||||||
|
shorthand: p
|
||||||
|
desc: Port USB
|
||||||
|
default: /dev/tty.usbmodem58760432961
|
||||||
|
cmds:
|
||||||
|
- python lerobot/scripts/configure_motor.py --port {{.PORT}} --brand feetech --model sts3215 --baudrate 1000000 --ID {{.ID}}
|
||||||
|
|
||||||
|
test-motor:
|
||||||
|
desc: Exécute les tests du moteur
|
||||||
|
cmds:
|
||||||
|
- python -m unittest {{.TESTS_DIR}}/test_motor_model.py
|
||||||
|
|
||||||
|
calibration:
|
||||||
|
desc: Exécute la calibration
|
||||||
|
cmds:
|
||||||
|
- python -m unittest {{.TESTS_DIR}}/test_display.py
|
||||||
|
|
||||||
|
find-usb-port:
|
||||||
|
desc: Trouve le port USB
|
||||||
|
cmds:
|
||||||
|
- python lerobot/scripts/find_motors_bus_port.py
|
||||||
|
|
||||||
|
teleoperate:
|
||||||
|
desc: Téléopère le robot
|
||||||
|
cmds:
|
||||||
|
- python lerobot/scripts/control_robot.py --robot.type=so100 --robot.cameras='{}' --control.type=teleoperate
|
Loading…
Reference in New Issue