使用训练脚本训练后保存模型文件并生成loss和准确率趋势图保存图片
训练代码如下
#coding: utf-8
import numpy as np
import matplotlib.pyplot as plt
import tensorflow as tf
from tensorflow.examples.tutorials.mnist import input_data
mnist = input_data.read_data_sets("MNIST_data", one_hot=True)
input_x = tf.placeholder(tf.float32, [None, 28*28])/255
output_y = tf.placeholder(tf.int32, [None, 10])
input_x_images = tf.reshape(input_x, [-1, 28, 28, 1])
#获取数据
test_x = mnist.test.images[:3000]
test_y = mnist.test.labels[:3000]
#构建网络
conv1 = tf.layers.conv2d(inputs=input_x_images,
filters=32,
kernel_size=[5, 5],
strides=1,
padding='same',