actpred: action_dim改18 + .mt脚本预训练入口
- 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
This commit is contained in:
1 parent
37206daf13
commit
45e019952e
6 files changed
+394
-12
No files matched your search
@@ -5,6 +5,9 @@ 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 "$@"
|
||||
|
||||
Reference in new issue
Block a user