
PyTorch
July_zh
越努力越幸运
展开
-
pytorch操作小技巧----模型性能观察
在我们创造一个新模型的时候,我们需要对一个模型进行全面的观察,与其他模型进行比较。下面就介绍几种观察模型性能的方法1查看模型的计算量和参数使用thop这个参数库from thop import profilefrom thop import clever_formatdef get_pf(model,input): flops, params = profile(model, inputs=(input,)) flops, params = clever_format([flop转载 2021-08-03 15:51:31 · 826 阅读 · 0 评论 -
动手学深度学习-pytorch
Introducing deep learning and the PyTorch libraryPyTorch是什么?为什么选择PyTorch?立即执行与延迟执行PyTorch是什么?PyTorch是一个Python程序库,有助于构建深度学习项目。PyTorch提供了一个核心数据结构Tensor(张量),这是一个多维数组,与NumPy数组有许多相似之处,在机器学习和其他数学密集型应用有广泛应用,不同之处在于张量可以使用GPU来加快计算。正如Python编程一样,PyTorch不仅是深入学习的入门,而且原创 2020-10-09 20:56:32 · 569 阅读 · 0 评论