Improve Error Message: Creating module attributes outside of setup()

[marcvanzee]

See #936 .

[avital]

The error isn't only if you assign to attributes inside @nn.compact -- it's at any time other than within setup

[marcvanzee]

Yes good point, updated the title.

[marcvanzee]

Duplicate of #847