分布式训练模型
代码在tensorflow\tf_v1\src目录下运行
-
先在终端输入
python multi_input_data.py命令生成tfrecords数据 -
开三个终端分别运行
python train.py --job_name=ps --task_index=0
python train.py --job_name=worker --task_index=0
python train.py --job_name=worker --task_index=1
-
Notifications
You must be signed in to change notification settings - Fork 1
luhao2013/distributed_train
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
分布式训练模型
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published