一般是torch save保存相关权重及训练参数
# 训练和测试循环 for epoch in range(start_epoch, epochs + 1): train(model, device, train_loader, optimizer, criterion, epoch) test(model, device