load_unet_weights

Load weights from TorchScript UNet into native UNet

Description

Load weights from TorchScript UNet into native UNet

Usage

load_unet_weights(native_unet, torchscript_path, verbose = TRUE)

Arguments

  • native_unet: Native UNet module
  • torchscript_path: Path to TorchScript UNet .pt file
  • verbose: Print loading progress

Value

The native UNet with loaded weights (invisibly)