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