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.