56 lines
1.5 KiB
Batchfile
56 lines
1.5 KiB
Batchfile
@echo off
|
||
chcp 65001 > nul
|
||
echo.
|
||
echo ╔════════════════════════════════════════════════════════════════╗
|
||
echo ║ d3rlpy 训练 - Unity 数据格式 ║
|
||
echo ╚════════════════════════════════════════════════════════════════╝
|
||
echo.
|
||
|
||
REM 检查 Python
|
||
python --version >nul 2>&1
|
||
if errorlevel 1 (
|
||
echo ✗ 未找到 Python
|
||
pause
|
||
exit /b 1
|
||
)
|
||
echo ✓ Python 已安装
|
||
|
||
REM 检查 d3rlpy
|
||
python -c "import d3rlpy" 2>nul
|
||
if errorlevel 1 (
|
||
echo.
|
||
echo ✗ 未安装 d3rlpy，正在安装...
|
||
pip install d3rlpy -i https://pypi.tuna.tsinghua.edu.cn/simple
|
||
if errorlevel 1 (
|
||
echo 安装失败，请手动安装: pip install d3rlpy
|
||
pause
|
||
exit /b 1
|
||
)
|
||
)
|
||
echo ✓ d3rlpy 已安装
|
||
|
||
REM 检查数据
|
||
if not exist "F:\TrainData\TrainingData\*.jsonl" (
|
||
echo.
|
||
echo ⚠️ 警告: 未找到训练数据
|
||
echo 请确保数据在: F:\TrainData\TrainingData\
|
||
echo.
|
||
)
|
||
|
||
REM 创建模型目录
|
||
if not exist "models" mkdir models
|
||
|
||
echo.
|
||
echo 开始训练...
|
||
echo ================================================================
|
||
echo.
|
||
|
||
python train.py
|
||
|
||
echo.
|
||
echo ================================================================
|
||
echo 训练结束
|
||
echo.
|
||
pause
|
||
|