- 博客(12)
- 收藏
- 关注
原创 pytorch中的loss function
1. CrossEntropyLoss() 和 NLLLoss()nn.CrossEntropyLoss() 由log_softmax()和NLLLoss()组成。import torch.nn.functional as Fimport torch.nn as nncriterion = nn.NLLLoss()# node_logits= Nx2 [[0,1], [1, 0], ...,[0,1]]node_logits = F.log_softmax(node_logits, dim=1
2022-03-08 13:13:16
1728
原创 python中的拼接操作
1. torch.tensor中的拼接我发现torch中stack没有list的append快。但是从一个大的嵌套的tensor中拿出几个重新组成一个新的tensor,可以用如下代码。不能直接append。poly_1 = torch.ones([3,5])poly_2 = torch.zeros([2,5])node_attr = torch.cat([poly_1, poly_2], dim=0)tmp_attr = node_attr.unsqueeze_(1)edge_attr = []
2022-01-04 19:33:45
821
原创 python加c++遍历路径下所有文件
1. 用python生成所有文件的路径并写入txt文档file_name = os.listdir(root)file_path = []path = os.path.join(root, file_name) + '\n'file_path.append(path)with open(txt_path, 'w+') as f: ...以上代码只是举例2. c++ 读取txt文件char line[200];char * txt_path= ;std::ifstream fin(t
2021-07-17 11:17:39
154
1
原创 small trick in python
1. 截取列tmp = torch.tensor([[1,2,3,4], [5,6,7,8]])cut = tmp[:, :2]截取tensor前面两列
2021-04-14 20:22:03
119
原创 python报错集锦
错误1: out = out + (h @ weight[i])RuntimeError: Expected object of scalar type Int but got scalar type Float for argument #3 'mat2' in call to _th_addmm_outh是torch.float类型 weight[i]是torch.int类型 @:matmul 会报错
2021-03-07 08:37:34
357
原创 os.list_dir排序
dirs = os.list_dir(root)dirs.sort(key= lamda x: int(x[:-4]))[1.txt, 2.txt, ....]因为没有对读取文件排序导致数组中错位。coding时总能发现我的智商不常上线。。。。。。
2021-03-05 19:19:38
329
原创 c++字符串和整数拼接
在生成路径时,我们需要将字符串和编号拼接。在python中format很容易生成,在c++中我花了半个小时。。。int i=0;char path[30];sprintf(path, "result/node/%d.txt", i);如果要传入函数,直接传入path就可以。因为字符数组首地址就是数组名。今天又被我机智到。...
2021-03-01 20:49:55
2552
1
原创 python3.7 安装gdsCAD
安装 gdsCAD 报错解决是因为get_version() 中有一句话, print get_version()这符合 python2 , 不符合 python3 的语法规则所以https://pypi.org/project/gdsCAD/#files 下载文件,之后修改一下git_version.py中的print ( get_version() )再在terminal中python setup.py installsuccessfully done!!...
2021-01-20 16:17:55
1230
1
原创 配置pytorch_geometric环境
要求是pytorch的版本大于等于1.4.0,因为我的电脑有一块GTX1650, 不想浪费,把原来的cuda 9.0 换成了 cuda 10.0(之前装过10.1)(发现不适配,也有可能是我当时装得有问题),装好了pytorch1.4.0+gpu 之后发现依赖包没有win版本的。放弃。。撞上了pytorch-1.6.0+cpu版本, torchvision-0.7.0版本。按着给的链接把所有依赖包装上, 大功告成。https://github.com/rusty1s/pytorch_geometric
2020-12-20 12:57:06
311
原创 Tensorflow安装总结,不使用anaconda
Tensorflow安装总结,不用anaconda虽然实验室里的gpu服务器可以跑tensorflow, 但因为我使用xshell不熟练,而且没法在pycharm上debug。我的电脑带GTX 1650显卡,不用白不用,虽然安装过程充满坎坷。之前发现因为没有总结,我忘记怎么装chrome插件的方法了,所以这回装的这么辛苦,我有必要总结一下,省的以后如果还要安装忘记。Anaconda下载失败网上大多方法都是通过anaconda安装,这次安装我起码卸了装了anaconda三四遍,因为anaconda默认下
2020-07-12 11:50:31
3896
4
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人