提交 02770a3d 编写于 作者: F frankwhzhang

train.py format and README

上级 368a7108
...@@ -74,17 +74,16 @@ python convert_format.py ...@@ -74,17 +74,16 @@ python convert_format.py
``` ```
## 训练 ## 训练
--use_cuda 表示使用gpu --parallel 表示多卡 '--use_cuda 1' 表示使用gpu --parallel 1 表示多卡
GPU 环境 GPU 环境
运行命令 `CUDA_VISIBLE_DEVICES=0 python train.py train_file test_file --use_cuda` 开始训练模型。 运行命令 `CUDA_VISIBLE_DEVICES=0 python train.py train_file test_file --use_cuda 1` 开始训练模型。
```
```python CUDA_VISIBLE_DEVICES=0 python train.py small_train.txt small_test.txt --use_cuda 1
CUDA_VISIBLE_DEVICES=0 python train.py small_train.txt small_test.txt --use_cuda
``` ```
CPU 环境 CPU 环境
运行命令 `python train.py train_file test_file` 开始训练模型。 运行命令 `python train.py train_file test_file` 开始训练模型。
```python ```
python train.py small_train.txt small_test.txt python train.py small_train.txt small_test.txt
``` ```
......
...@@ -185,6 +185,7 @@ def train_net(): ...@@ -185,6 +185,7 @@ def train_net():
test_file = args.test_file test_file = args.test_file
use_cuda = True if args.use_cuda else False use_cuda = True if args.use_cuda else False
parallel = True if args.parallel else False parallel = True if args.parallel else False
print("use_cuda:", use_cuda, "parallel:", parallel)
batch_size = 50 batch_size = 50
vocab, train_reader, test_reader = utils.prepare_data( vocab, train_reader, test_reader = utils.prepare_data(
train_file, test_file,batch_size=batch_size * get_cards(args),\ train_file, test_file,batch_size=batch_size * get_cards(args),\
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册