Skip to content

Add TorchScript support for networks #1160

@Nic-Ma

Description

@Nic-Ma

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.

Metadata

Metadata

Assignees

Labels

enhancementNew feature or request

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions