OUSMLoss is defined as an nn.Module, while .backward() is a tensor method.
You would either have to implement the backward() method in this module or call .backward() on the loss tensor (probably the return tensor).