AI教程网 - 未来以来,拥抱AI;新手入门,从AI教程网开始......

 分类:PyTorch手册

PyTorch 是一个 Python 优先的深度学习框架,由Facebook 的 AI 研究团队发布。

PyTorch手册

PyTorch:序列化语义

PyTorch:序列化语义
最佳实践 保存模型的推荐方法 这主要有两种方法序列化和恢复模型。 第一种(推荐)只保存和加载模型参数: torch.save(the_model.state_dict(), PATH) 然后: the_model = TheModelCl...

AI君 5年前 (2019-03-28) 135℃ 2喜欢

PyTorch手册

PyTorch:扩展PyTorch

PyTorch:扩展PyTorch
本篇文章中包含如何扩展 torch.nn, torch.autograd和 使用我们的 C 库编写自定义的C扩展。 扩展 torch.autograd 如果你想要添加一个新的 Operation 到autograd的话,你的Operatio...

AI君 5年前 (2019-03-28) 149℃ 2喜欢

PyTorch手册

PyTorch:多进程最佳实践

PyTorch:多进程最佳实践
torch.multiprocessing是Pythonmultiprocessing的替代品。它支持完全相同的操作,但扩展了它以便通过multiprocessing.Queue发送的所有张量将其数据移动到共享内存中,并且只会向其他进程发送...

AI君 5年前 (2019-03-28) 140℃ 10喜欢

PyTorch手册

PyTorch:自动求导机制

PyTorch:自动求导机制
本说明将概述Autograd如何工作并记录操作。了解这些并不是绝对必要的,但我们建议您熟悉它,因为它将帮助您编写更高效,更简洁的程序,并可帮助您进行调试。 从后向中排除子图 每个变量都有两个标志:requires_grad和volatile...

AI君 5年前 (2019-03-28) 115℃ 2喜欢

PyTorch手册

PyTorch:CUDA语义

PyTorch:CUDA语义
torch.cuda会记录当前选择的GPU,并且分配的所有CUDA张量将在上面创建。可以使用torch.cuda.device上下文管理器更改所选设备。 但是,一旦张量被分配,您可以直接对其进行操作,而不考虑所选择的设备,结果将始终放在与张...

AI君 5年前 (2019-03-28) 143℃ 2喜欢

PyTorch手册

PyTorch中文文档

PyTorch中文文档
PyTorch是使用GPU和CPU优化的深度学习张量库。 说明 自动求导机制 CUDA语义 扩展PyTorch 多进程最佳实践 序列化语义 Package参考 torch torch.Tensor torch.Storage torc...

AI君 5年前 (2019-03-28) 285℃ 3喜欢