image.png1263×401 59.8 KB

Hi, Actually I am trying to train skip-thought vectors on multiple-gpus, this implementation is taken from  https://github.com/sanyam5/skip-thoughts 3  code. But, while computing loss function, am getting above error. Can anybody help me to resolve this issue. Thank you.
Loss snippet code:

image.png1247×373 53.8 KB