AttributeError: 'ReLU' object has no attribute 'weight'

[SherlockHua1995]

Hello, mightydeveloper. When I use 'weight_share.py' to compress the trained model, the error occured: AttributeError: 'ReLU' object has no attribute 'weight' . File "weight_share.py", line 32, in apply_weight_sharing(model) File "/net/quantization.py", line 16, in apply_weight_sharing dev = module.weight.device File "/usr/local/lib/python3.5/dist-packages/torch/nn/modules/module.py", line 539, in getattr type(self).name, name)) AttributeError: 'ReLU' object has no attribute 'weight' How to jump over the ReLU module? THANKS.