TensorFlow MNIST示例
先决条件
TensorFlow 2.2.0
训练
$ python train.py \
--train_batch_size=64 \
--test_batch_size=50 \
--epochs=5 \
--output_dir="./models"
评价
$ python evaluate.py \
--model_dir="./models" \
--batch_size=50
预测
$ python predict.py \
--model_dir="./models" \
--image=<path>
运行gRPC服务器
从digit_classification.proto生成Python代码,该代码提供gRPC服务器和客户端接口。
$ python -
评论0
最新资源