feat(actpred): 动作预测模块入库(PyTorch 训练/预测/ONNX 导出)

This commit is contained in:
qing committed 2026-08-08 06:25:06 +00:00
1 parent f93ae59521
commit 575c2a27a4
11 files changed
+752

No files matched your search

+12
View File
@@ -0,0 +1,12 @@
#!/bin/bash
cd "$(dirname "$0")" || exit
python3 -m venv venv
source venv/bin/activate
# CPU-only PyTorch (no CUDA dependency)
pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu
# Remaining deps from Tsinghua mirror
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
mkdir -p checkpoints data