weixin_42155821的博客

图神经网络debug

pytorch中文文档阅读笔记 CUDA语义与实践 通过torch.cuda.device(n)来控制使用哪一块GPU，可以采用with调用，也可采用tensor.device调用 由于在固定内存（页锁）中的数据，在主机和GPU之间的复制速度快得多。因此CPU张量和存储开放了一个pin_memory()方法，将它的数据放在固定区域 torch 常用函数 运算part torc...

NLLLoss与CE Adam优化算法 原址(https://www.cnblogs.com/yifdu25/p/8183587.html) 是一阶优化算法，全称适应性矩估计（adaptive moment estimation）。在非凸优化问题中有如下优势： 计算高效 内存需要的少 梯度对角缩放不变 适用于大规模数据...