请叫我杨先生

摘要： Pytorch GPU 加速 (拥有Nvidia显卡) 查看自己的 torch 版本型号 >>> torch.__version__ '1.10.1+cpu' 假如你的 torch 版本是和我一样的，那么你就需要安装 CUDA11.1的版本 对应表如下：\(CLICK HERE\) 下载安装CUDA 阅读全文

摘要： 自定义层 早在Pytorch 5.1 中我们就有讲过自定义层的实现和原理，具体可以参考我的前两篇文章。 这篇文章主要是补充带参数的层的自定义，和不带参数的层的自定义。 不带参数自定义层 class CenterLayer(nn.Module): # 我们定义其为 a-a的均值 def __init_ 阅读全文