今天看啥  ›  专栏  ›  大模型智能

十分钟掌握Pytorch搭建神经网络的流程

大模型智能  · 公众号  ·  · 2024-06-13 23:40
    

文章预览

大模型智能|分享 来源 | 古月居 最近发现身边的一些初学者朋友捧着各种pytorch指南一边看一边敲代码,到最后反而变成了打字员。 敲完代码一运行,出来结果和书上一对比,哦,是书上的结果,就翻到下一章。 半天就能把一本书都打完,但是合上书好像什么都不记得。有的甚至看了两三遍,都搭不出一个简单的网络来,这种学习方式很不可取。 如果你刚好是这种情况,这篇文章应该能给你一些帮助。如果你已经是进阶的水平了,就直接关掉页面就好了。 pytorch的网络搭建,比tensorflow简单很多。格式很好理解。 如果你想做一个网络,需要先定义一个Class,继承 nn.Module(这个是必须的,所以先import torch.nn as nn,nn是一个工具箱,很好用),我们把class的名字就叫成Net. Class Net (nn.Module) : 这个Class里面主要写两个函数,一个是初始化的__init__函数, ………………………………

原文地址:访问原文地址
快照地址: 访问文章快照
总结与预览地址:访问总结与预览