Add the missing kwargs for torch.jit.script

[Nic-Ma]

Is your feature request related to a problem? Please describe.
Currently, we defined kwargs in the convert_to_torchscript:
https://github.com/Project-MONAI/MONAI/blob/dev/monai/networks/utils.py#L441
But forgot to add it in the call of torch.jit.script and add the missing doc-string.
Would be nice to add it.