文件名称:tensorflow-mnist:TensorFlow MNIST示例PyTorch的实现是https:github.comghwnpytorch-mnist
文件大小:4KB
文件格式:ZIP
更新时间:2024-03-03 21:59:25
Python
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=
【文件预览】:
tensorflow-mnist-master
----predict.py(1KB)
----README.md(954B)
----server.py(1KB)
----train.py(2KB)
----digit_classification.proto(182B)
----evaluate.py(768B)
----requirements.txt(52B)