随笔分类 - pytorch
摘要:import torch torch.manual_seed(2021) # in_channels 表示输入特征数量,卷积核的第一个维度 # out_channels 表示输出特征数量,也是卷积核的数量 # kernel_size 是卷积核的第二维度。 卷积核维度为 in_channels * k
阅读全文
摘要:torch.equal();该方式判断tensor,形状和值是否完全相等。不太适用于浮点 torch.allclose();该方式判断tensor,形状和值是否足够接近,适用于浮点对比 torch.set_printoptions(precision=10) 打印设置;
阅读全文
摘要:安装方法 pytorch-geometric 不支持同时多个版本ptorch, 而conda 默认装cpu gpu 两个版本,所以需要修改默认配置, 添加 cpuonly参数等。或使用pip安装单一版本pyroch 参考 https://github.com/pyg-team/pytorch_geo
阅读全文

浙公网安备 33010602011771号