Siddhartha

@Ahmux2018 Thanks. It seems we also have to set shape parameter to 'image' otherwise the image isn't displayed.

Has anyone managed to fix this ? `lalsuite==6.82` `pycbc==1.18.1` `pycbc.noise.reproduceable` doesn't seem to work for me, the jupyter notebook just restarts without throwing any error or warning.

Runtime > NAME="Ubuntu" VERSION="18.04.5 LTS (Bionic Beaver)" ID=ubuntu ID_LIKE=debian PRETTY_NAME="Ubuntu 18.04.5 LTS" VERSION_ID="18.04" HOME_URL="https://www.ubuntu.com/" SUPPORT_URL="https://help.ubuntu.com/" BUG_REPORT_URL="https://bugs.launchpad.net/ubuntu/" PRIVACY_POLICY_URL="https://www.ubuntu.com/legal/terms-and-policies/privacy-policy" VERSION_CODENAME=bionic UBUNTU_CODENAME=bionic List of packages installed in my environment [pip_list.txt](https://github.com/gwastro/pycbc/files/6764474/pip_list.txt) If i use...

Sure, thanks!

Putting here for future reference, I got this error (on a different machine). Using the dev version and `pycbc.noise.reproduceable` seems to work. However, `pycbc.noise.noise_from_psd` gives error even with the dev...

Thanks! I got the same error in ubuntu 20.04. Your solution worked!

@Vegeta2020 what do you think ?

It will be nice if this PR can be merged.

@passionvery You may want to check if you are using `F.normalize` before calculating the loss function [(like here](https://github.com/HobbitLong/SupContrast/blob/6d5a3de39070249a19c62a345eea4acb5f26c0bc/networks/resnet_big.py#L185)). I missed this step and got nan.

@qq456cvb Any update about the performance after the change ?