Browse Source

fix bug

pull/3/head
liuzx 2 years ago
parent
commit
5ec2998703
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      gpu_mnist_example/train.py

+ 1
- 1
gpu_mnist_example/train.py View File

@@ -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


Loading…
Cancel
Save