随笔分类 - Pytorch
摘要：CUDA是一个并行计算平台和编程模型，能够使得使用GPU进行通用计算变得简单和优雅。Nvidia官方提供的CUDA 库是一个完整的工具安装包，其中提供了 Nvidia驱动程序、开发 CUDA 程序相关的开发工具包等可供安装的选项。 Nvidia CUDA Nvidia安装的cuda是一个完整的cud
阅读全文
摘要：import torch.nn as nn torch.nn.Sigmoid() 一、sigmoid介绍 sigmoid是激活函数的一种，它会将样本值映射到0到1之间。 sigmoid的公式如下：
摘要：转自 https://cloud.tencent.com/developer/article/1710864 1. 引言 在PyTorch中，对模型参数的梯度置0时通常使用两种方式：model.zero_grad()和optimizer.zero_grad()。二者在训练代码都很常见，那么二者的区别
阅读全文
