添加训练模型保存

main.py

@@ -175,6 +175,6 @@ def net_eval():
 time_cb = TimeMonitor(data_size=60)
 loss_cb = LossMonitor()
 model = Model(net, net_loss, net_opt, metrics={"Accuracy": Accuracy()})
-model.train(10, train_dataset, callbacks=[time_cb, loss_cb], dataset_sink_mode=True)
+model.train(10, train_dataset, callbacks=[time_cb, loss_cb, ckpoint], dataset_sink_mode=True)
 
 net_eval()
\ No newline at end of file