提交 ad9ff0b4 编写于 作者: F frankwhzhang

fix Readme & train.py

上级 02770a3d
......@@ -74,7 +74,7 @@ python convert_format.py
```
## 训练
'--use_cuda 1' 表示使用gpu --parallel 1 表示多
'--use_cuda 1' 表示使用gpu, 缺省表示使用cpu '--parallel 1' 表示使用多卡,缺省表示使用单
GPU 环境
运行命令 `CUDA_VISIBLE_DEVICES=0 python train.py train_file test_file --use_cuda 1` 开始训练模型。
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册