nn.DataParallel will call the wrapped module directly (__call__ first, then inside it forward).
If you call some custom methods from the model, data parallel won’t be used, so instead call your model directly again.