© 1999-2048 dssz.net 粤ICP备11031372号
[其它] torchextractor:torchxtractor使特征提取变得简单-源码
说明:割炬:PyTorch中间特征提取 介绍 仅仅因为forward功能未返回用户期望的内容,许多模型定义就被无情地复制粘贴了。 使用torchextractor您可以轻松捕获中间特征图,并在其上构建任何您喜欢的东西。 安装 pip install git+https://github.com/antoinebrl/torchextractor.git 用法 import torch import torchvision import torchextractor as tx model = torc<weixin_42100971> 在 上传 | 大小:13312