这篇博客将介绍如何使用PyTorch深度学习库训练第一个神经网络。具体来说是如何在scikit learn的make_blobs函数生成的示例数据集上训练PyTorch神经网络。
PyTorch库初学者可能遇到的最大错误是忘记或混淆了以下步骤:
PyTorch库功能强大,但使用PyTorch训练神经网络就像脱下自行车的训练轮——如果混淆了重要步骤,就没有安全网的保护了(不像Keras/TensorFlow,它允许将整个训练过程封装到一个model.fit调用中。
这并不是说Keras/TensorFlow比PyTorch“更好”,这只是两个深度学习库之间的区别。
前几行输出显示了简单的4
上一篇:牛客竞赛每日俩题 - Day14