AttributeError: ‘Conv2d’ object has no attribute ‘total_ops’
File "/home/...../..._encoder.py", line 34,in forward
x = self.conv(x)
File "/home/...../python3.8/site-packages/torch/nn/modules/module.py", line 1511,in _wrapped_call_impl
return self._call_imp