Is your feature request related to a problem? Please describe.
Now we already fixed some TorchScript issues for several networks and added unit tests for them.
Still need to resolve TorchScript issues in other networks: SegResNet & SegResNetVAE, DenseNet, SENet, DynUNet.
Describe the solution you'd like
torch.jit.script() can work for these networks.