文件名称:torchfunc:PyTorch功能和实用程序使您的生活更轻松
文件大小:1.35MB
文件格式:ZIP
更新时间:2024-05-24 17:32:41
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