weixin_42727538的博客

使用anaconda为环境安装对应的cuda版本

下载地址：https://download.pytorch.org/whl/torch_stable.html

在加载模型时出现 AttributeError: 'str' object has no attribute 'copy' 错误解决方案：将model.load_state_dict(state_dict) 修改为：model.load_state_dict(torch.load(PATH))

最近需要使用到PyG框架，安装的时候需要注意一些问题，记录一下，方便后来者避坑！步骤1首先要先确定自己的torch版本如果使用的Anaconda可以使用conda list命令查看版本号进入官网https://pytorch-geometric.com/whl/找到与torch对应的torch-geometric，cu表示cuda版本，cpu表示cpu版本，选择对应的进去。cuda版本号可以在控制台使用nvcc --version来查看![在这里插入图片描述](https://img-blog

nll_losstorhc.nn.functional中给出的关于nll_loss的一个样例target: 数值 <= C-1, 一般可以看作一个一维列表，存放的是真实类别编号先经过softmax，再经过log，最后再使用nll_loss# input is of size N x C = 3 x 5input = torch.randn(3, 5, requires_grad=True)# each element in target has to have 0 <= value

最近在基于vgg的网络模型上进行改进，记录下几种常见的vgg模型结构：vgg11，vgg13，vgg16，vgg19vgg11VGG-11的网络结构： VGG( (features): Sequential( (0): Conv2d(3, 64, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1)) (1): ReLU(inplace=True) (2): MaxPool2d(kernel_size=2, stride=2,

m个样本的梯度下降其中J是成本函数，相关定义如下dw1和dw2是各项损失对参数w1的导数我们在这里假设只有w1和w2两个参数，同理可扩展为m个样本（使用for循环）最后在遍历结束之后，将成本函数和各个参数除以m，就得到了m个样本的成本函数和参数的导数...