pytorch
|晴天|
这个作者很懒,什么都没留下…
展开
-
python实现召回率、精度、f1代码
import matplotlib.pyplot as pltimport numpy as np# 读取实验结果中的精度和损失def data_plot(path): # path = r'./model_png\5gcn_128_node_pos_floor_2000.log' # path = r'./[6, 6]_Standard_taz.csv' with open(path, mode="r", encoding="utf-8") as f: da原创 2020-11-03 11:11:10 · 10279 阅读 · 1 评论 -
pytorch运行错误:RuntimeError: cuDNN error: CUDNN_STATUS_INTERNAL_ERROR
发现了解决该问题的方法:(特别是实验室服务器单机多卡的情况)TensorFlow时代的做法是,import osos.environ['CUDA_ENABLE_DEVICES'] = '0'但是该方法在**Pytorch**代码并不适用,正确的操作方法是:import torchtorch.cuda.set_device(0)————————————————...原创 2020-06-26 20:14:48 · 10567 阅读 · 6 评论 -
RuntimeError: Expected object of scalar type Double but got scalar type Float for argument #2 'mat2'
错误异常代码:def build_adjacency(self, g): 。。。。。。。。。(部分代码省略) adjacency = np.zeros([len(adj_dict), len(adj_dict)]) for key, value in adj_dict.items(): for item in value: ...原创 2020-05-05 22:13:53 · 306 阅读 · 0 评论 -
mnist数据集下载(附百度网盘链接永久有效)
https://github.com/amplab/datascience-sp14/raw/master/lab7/mldata/mnist-original.mat原创 2020-04-21 13:50:47 · 7031 阅读 · 2 评论 -
PyTorch什么是nn.relu()与F.relu()之间的区别
什么是**nn.relu()与nn.functional.relu()**之间的区别,如果没有区别那么为什么会出现这种重复…?nn.ReLU()创建一个nn.Module,您可以将其添加到例如nn.Sequential模型中。另一方面,nn.functional.relu只是对relu函数的函数API调用,因此你可以自己添加它,例如在你的forward方法中。一般来说,如果你的话,它可能取决于你...原创 2019-08-08 21:21:43 · 7016 阅读 · 0 评论 -
torch.multinomial()理解
torch.multinomialtorch.multinomial(input, num_samples,replacement=False, out=None) → LongTensor返回一个张量,每行包含从input相应行中定义的多项分布中抽取的num_samples个样本。**[注意]:**输入input每行的值不需要总和为1 (这里我们用来做权重),但是必须非负且总和不能为0。...原创 2019-08-23 09:30:41 · 1855 阅读 · 0 评论 -
torch.manual_seed(args.seed)的作用
# 为CPU设置种子用于生成随机数,以使得结果是确定的torch.manual_seed(args.seed)# torch.cuda.manual_seed()为当前GPU设置随机种子if args.cuda: torch.cuda.manual_seed(args.seed)# 如果使用多个GPU,应该使用torch.cuda.manual_seed_all()为所有的GP...原创 2019-09-18 09:53:16 · 8268 阅读 · 0 评论 -
OSError: [WinError 126] 找不到指定的模块
Traceback (most recent call last): File "D:/coding/dgl_master/dgltest.py", line 2, in <module> import dgl File "D:\software\python3\lib\site-packages\dgl\__init__.py", line 6, in <mod...原创 2019-10-17 15:21:13 · 8397 阅读 · 11 评论 -
pytorch中tensor张量数据类型的转化
1.tensor张量与numpy相互转换tensor ----->numpyimport torcha=torch.ones([2,5])tensor([[1., 1., 1., 1., 1.], [1., 1., 1., 1., 1.]])# ********************************** b=a.numpy()arra...原创 2019-09-29 16:57:13 · 21699 阅读 · 0 评论