请问一下有训练模型并保存成pkl文件的代码吗

[17839792633]

请问一下有训练模型并保存成pkl文件的代码吗

[wyc941012]

hello，代码中已经给出了alexnetlayer.py和vgg16layer.py的分层模型结构，训练集在datasets/cifar_10目录，这块可以参考下网上PyTorch的用法如何迭代的训练神经网络，最后保存模型成为pkl文件。抱歉因为之前的训练代码确实找不到了，我就是按照一般的写法来训练的跟网上常用的应该一致。