PyTorch - 神经网络到功能块

训练深度学习算法涉及以下步骤 −

  • 构建数据管道
  • 构建网络架构
  • 使用损失函数评估架构
  • 使用优化算法优化网络架构权重

训练特定的深度学习算法是将神经网络转换为功能块的确切要求,如下所示 −

功能块

关于上图,任何深度学习算法都涉及获取输入数据,构建各自的架构,其中包括嵌入其中的一堆层。

如果您观察上图,则使用损失函数对神经网络权重的优化进行评估。