From 993ca4d97b48d5795628532bc79d8baa87709956 Mon Sep 17 00:00:00 2001 From: hexiangdong2019 <PIpi589632147> Date: Sun, 8 Aug 2021 15:49:03 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E8=AE=AD=E7=BB=83=E5=8F=82?= =?UTF-8?q?=E6=95=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/main.py b/main.py index 59e7c5a..8391d0a 100644 --- a/main.py +++ b/main.py @@ -178,7 +178,7 @@ def net_eval(): time_cb = TimeMonitor(data_size=60) loss_cb = LossMonitor() model = Model(net, net_loss, net_opt, metrics={"Accuracy": Accuracy()}) -model.train(500, train_dataset, callbacks=[time_cb, loss_cb, ckpoint], dataset_sink_mode=True) +model.train(100, train_dataset, callbacks=[time_cb, loss_cb, ckpoint], dataset_sink_mode=True) net_eval() -- GitLab