search

Sanster / pytorch-network-slimming

A package to make do Network Slimming a little easier
47 stars 10 forks source link

python3 gen_schema.py --net resnet18 --save_path ./schema/resnet18.json 报错 #8

Closed GeneralJing closed 2 years ago

GeneralJing commented 2 years ago

Traceback (most recent call last): File "gen_schema.py", line 31, in y = model.features(x) File "/home/xxx/anaconda3/envs/slim/lib/python3.8/site-packages/torch/nn/modules/module.py", line 778, in getattr raise ModuleAttributeError("'{}' object has no attribute '{}'".format( torch.nn.modules.module.ModuleAttributeError: 'ResNet' object has no attribute 'features'

Sanster commented 2 years ago

torchvision 的版本？

GeneralJing commented 2 years ago

torch=1.7.0 torchvision=0.8.1 是用的requirements安装的

Sanster commented 2 years ago

调试用的代码忘记删了，已修复：https://github.com/Sanster/pytorch-network-slimming/commit/e3fd71d3d2d21b00acc03249cc83f15c7161a077