网络模型创建的步骤及nn.Module属性

  在数据模块中我们了解了pytorch是如何从硬盘中读取数据,然后对数据进行预处理,最后转化为张量的形式输入到我们的模型当中。   在模型里,是对数据模块出处的张量进行复杂的数学运算,最后得到可以分类、分割、目标检测的输出。     一、网络模型创建的步骤 这一节学习模型模块。     模型模块又分两个模块,一个是模型创建,一个是权值初始化。 在模型创建中,又分为两个子模块。构建网络层和拼接网络
相关文章
相关标签/搜索