修改训练模型保存设置

main.py

@@ -178,3 +178,4 @@ model = Model(net, net_loss, net_opt, metrics={"Accuracy": Accuracy()})
 model.train(10, train_dataset, callbacks=[time_cb, loss_cb, ckpoint], dataset_sink_mode=True)
 
 net_eval()
+