tensorflow2 mnist github上的

先贴上我的环境，环境很重要

1 进入新创建的计算环境： 打开 Anaconda Prompt
2 查看所有计算环境

cona env list

3 进入新建的tensorflow2的环境

conda activate tensorflow2

4 查看所有的包的版本号

pip list

下载 mnist 代码

mnnist相关代码在 models-master\official\vision\image_classification文件中， 进入代码所在路径

第一次跑代码，要加上--download 参数，下载数据集

python mnist_main.py --model_dir=E:\AI\mnist\models --data_dir=E:\AI\mnist\data --download --train_epochs=10

参数--model_dir一定要设置，不然跑不过。--data_dir参数若是不设置,数据会保存在根目录的 tmp文件夹中

可以看到在训练了。

要注意，在 mnist_main.py 代码中加入

import sys
sys.path.append('..\\..\\..\\..\\models-master')

不然会报找不到 official 模块

在 tensorboard 中查看可视化图

tensorboard --logdir E:\AI\mnist\models

--logdir 的值填的就是上面模型存放的目录，代码中看出来的

  callbacks = [
      tf.keras.callbacks.ModelCheckpoint(
          ckpt_full_path, save_weights_only=True),
      tf.keras.callbacks.TensorBoard(log_dir=flags_obj.model_dir),
      
  ]

在浏览器中输入 http://localhost:6006/， 查看界面