随笔分类 -  Pytorch

摘要:转自 https://blog.csdn.net/mao_jonah/article/details/89502380 许多Python项目中都包含了requirements.txt文件,该文件记录了当前程序的所有依赖包及其精确版本号。 生成requirement.txt文件 pip freeze 阅读全文
posted @ 2021-01-26 21:24 Gelthin 阅读(10846) 评论(0) 推荐(0)
摘要:Huawei MateBook 14 2020 款安装 Pytorch 发现似乎原本 cuda10.2 已经默认安装好,没有独立安装 cuda 安装包,不知道后面是否会出问题。 直接按照官网命令 conda install pytorch torchvision torchaudio cudatoo 阅读全文
posted @ 2020-11-26 14:22 Gelthin 阅读(1023) 评论(0) 推荐(0)
摘要:一般而言,可以按照如下方式固定随机数种子,以便复现实验: # 来自相关于 GCN 代码: 例如 grand.py 等的代码 parser.add_argument('--seed', type=int, default=42, help='Random seed.') np.random.seed( 阅读全文
posted @ 2020-11-25 11:03 Gelthin 阅读(1036) 评论(0) 推荐(0)
摘要:最近准备复现一下 KDD-20 Towards Deeper Graph Neural Networks 的代码,顺便学习一下 GCN 最新的一些进展。 GCN 的代码通常需要安装 Pytorch on Graph, 按照其 官方指导, 发现无法兼容当前所使用的服务器上的 pytorch1.3 和 阅读全文
posted @ 2020-10-31 18:05 Gelthin 阅读(148) 评论(0) 推荐(0)
摘要:2020.10.1 发现最新的 pytorch 已经原生支持 one-hot 操作 torch.nn.functional.one_hot()pytorch 官方文档链接 只需如下一行代码: label_one_hot = torch.nn.functional.one_hot(labels, se 阅读全文
posted @ 2020-09-16 19:19 Gelthin 阅读(4100) 评论(0) 推荐(1)
摘要:1. 大幅度提升 Pytorch 的训练速度 device = torch.device("cuda" if torch.cuda.is_available() else "cpu") torch.backends.cudnn.benchmark = True 但加了这一行,似乎运行结果可能会存在细 阅读全文
posted @ 2020-01-04 09:35 Gelthin 阅读(1878) 评论(0) 推荐(0)
摘要:1. Colthing1M 数据集中有的图片没有 224*224大, 直接用 transforms.RandomCrop(224) 就会报错,RandomRange 错误 raise ValueError("empty range for randrange() (%d,%d, %d)" % (is 阅读全文
posted @ 2019-12-13 09:39 Gelthin 阅读(1737) 评论(0) 推荐(0)
摘要:参考 https://blog.csdn.net/weixin_42401701/article/details/80820778 和 https://www.cnblogs.com/lllcccddd/p/10661966.html 一些相关的命令 conda update -n base con 阅读全文
posted @ 2019-11-26 22:49 Gelthin 阅读(5186) 评论(0) 推荐(0)
摘要:2019.11.16 为了使用远程的云服务器,必须要自己配置环境,这次还算比较顺利。 1. 安装cuda https://blog.csdn.net/wanzhen4330/article/details/81699769 ( 安装cuda = nvidia driver + cuda toolki 阅读全文
posted @ 2019-11-16 23:10 Gelthin 阅读(1063) 评论(0) 推荐(0)