qing
45e019952e
- config.py: action_dim默认值 8→18（对齐bot4cj 18列） - mt_parser.py: 轻量级.mt @Action解析器（支持-占位符前向填充） - pretrain_mt.py: 预训练入口（零图像占位，checkpoint兼容现有推理） - _run.sh: 新增pretrain子命令 - README.md: 新增预训练文档 + action_dim说明 - test_actpred.py: 测试断言更新为18
17 lines
349 B
Bash
Executable File
17 lines
349 B
Bash
Executable File
#!/bin/bash
|
|
cd "$(dirname "$0")" || exit
|
|
source ./venv/bin/activate
|
|
|
|
if [ "${1:-}" = "train" ]; then
|
|
shift
|
|
exec python3 train.py "$@"
|
|
elif [ "${1:-}" = "pretrain" ]; then
|
|
shift
|
|
exec python3 pretrain_mt.py "$@"
|
|
elif [ "${1:-}" = "export" ]; then
|
|
shift
|
|
exec python3 export_onnx.py "$@"
|
|
else
|
|
exec python3 predict.py "$@"
|
|
fi
|