From 5ec299870393eea06fe7e16a4af4f34408eca74a Mon Sep 17 00:00:00 2001 From: liuzx Date: Wed, 31 Jan 2024 17:14:31 +0800 Subject: [PATCH] fix bug --- gpu_mnist_example/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/gpu_mnist_example/train.py b/gpu_mnist_example/train.py index 9db7665..e71eaf2 100644 --- a/gpu_mnist_example/train.py +++ b/gpu_mnist_example/train.py @@ -109,7 +109,7 @@ if __name__ == '__main__': start_epoch = 0 print('无保存模型，将从头开始训练！') - for epoch in range(start_epoch+1, epochs): + for epoch in range(start_epoch+1, epochs+1): train(model, train_loader, epoch) test(model, test_loader, test_dataset) # 将模型保存到c2net_context.output_path