Patchcore export model very slow

[kanoon1991]

From I training Patchcore model by Anomalib version 2.3.0dev with half precision = PrecisionType.FLOAT16, I found on step export model very slow, How about we can improve export model speed?

GPU model : RTX4090 24GB, PyTorch version: 2.8.0+cu129, CUDA version (PyTorch): 12.9

Training code

from anomalib.data import Folder
from anomalib.models import Patchcore
from anomalib.engine import Engine
from anomalib.deploy import ExportType, CompressionType
from anomalib.pre_processing import PreProcessor
from anomalib import PrecisionType
from torchvision.transforms.v2 import Compose, Resize
import multiprocessing as mp

def main():
train_transform = Compose([
Resize(512),
])

datamodule = Folder(
    root=r"datasets\0",
    name="0_Patchcore",
    normal_dir="good",
    abnormal_dir="abnormal",
    train_batch_size=8,
    eval_batch_size=8,
    num_workers=8,
)

pre_processor = PreProcessor(transform=train_transform)

model = Patchcore(
    backbone="wide_resnet50_2",
    pre_processor=pre_processor,
    coreset_sampling_ratio=0.1,
    precision= PrecisionType.FLOAT16,
)


engine = Engine(
    accelerator="auto",
    devices=1,
)

engine.fit(datamodule=datamodule, model=model)
engine.test(datamodule=datamodule, model=model, ckpt_path=None, verbose=True)
engine.export(model=model, export_type=ExportType.ONNX)

if name == "main":
mp.freeze_support()
main()