【文件属性】:
文件名称:torchfunc:PyTorch功能和实用程序使您的生活更轻松
文件大小:1.35MB
文件格式:ZIP
更新时间:2021-05-04 23:46:01
docker performance utilities neural-network extensions
改善和分析神经网络的性能(例如Tensor Cores兼容性)
在数据通过时记录/分析torch.nn.Module内部状态
根据外部条件执行上述操作(使用单个Callable进行指定)
日常神经网络相关职责(模型大小,播种,时间测量等)
获取有关您的主机操作系统, torch.nn.Module设备,CUDA功能等的信息。
版本
文件
测验
覆盖范围
风格
聚酰亚胺
Python
火炬
码头工人
路线图
:light_bulb: 例子
在此处检查文档: :
1.获取性能提示
获取有关模块的即时性能提示。 评论所描述的所有问题都将通过torchfunc.performance.tips :
class Model ( torch . nn . Module ):
def __init__ ( self ):
super (). __init__ ()
se