project-monai's picture
Upload brats_mri_generative_diffusion version 1.1.3
e217a55 verified
raw
history blame contribute delete
248 Bytes
{
"+imports": [
"$from monai.networks import trt_compile"
],
"diffusion": "$trt_compile(@network_def.to(@device), @load_diffusion_path)",
"autoencoder": "$trt_compile(@autoencoder_def.to(@device), @load_autoencoder_path)"
}