Pytorch

最新推荐文章于 2025-03-18 16:01:21 发布

原创最新推荐文章于 2025-03-18 16:01:21 发布 · 266 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#PyTorch #noob

PyTorch作为深度学习框架的后起之秀，因其强大而简单的集成模块圈粉无数。为便于新人入坑，PyTorch官网上有一个简短的入坑教程。在这里说一说这个教程里我认为值得注意的两个点。
1 并非仅仅GPU
这个点很简单，就是PyTorch在CPU下也能跑，这就给想学习深度学习但是手上暂时没有GPU的小伙伴提供了绿色通道。
2 你能干什么
在入坑教程的Training a classifier部分，里面讲了他们训练一个分类器的5个步骤：
1. Load and normalizing the CIFAR10 training and test datasets using torchvision
2. Define a Convolution Neural Network
3. Define a loss function
4. Train the network on the training data
5. Test the network on the test data
其实个流程我认为指出了我们能够方便地控制的两点，即以Define开头的两点：定义神经网络和定义损失函数。因为其它的PyTorch已经帮我们做好了。感觉好像在搭积木。