Loading

随笔分类 -  pytorch

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