Я использую факельную модель, чтобы сделать вывод. Во время выполнения логического вывода я заметил, что нормализация тензорного изображения потребляет половину времени логического вывода.
Функция, которая вызывается:
input_image_normalizer = Compose([
ToTensor(),
Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]),
])
При удалении функции нормализации логическое значение логического вывода логического вывода равно в два раза.
Можно ли провести какую-либо оптимизацию, чтобы уменьшить время операции нормализации? Я подумываю над тем, чтобы поставить функцию нормализации на графический процессор? Если так, то как это можно сделать? У меня есть CUDA и torchvision.
Функция нормализации: https://pytorch.org/docs/stable/torchvision/transforms.html#torchvision .transforms.Normalize