Совершенно новый в машинном обучении - у меня есть обучающий скрипт, который обучает модель по умолчанию без весов в классификаторе, используя изображения в качестве входных данных, и после обучения он выдает выходные данные с файлом .hdf5.
Мне известно, что файл .hdf5 будет содержать информацию о модели и весах (https://www.tinymind.com/learn/terms/hdf5).) Но я пытаюсь найти способ использовать выходной файл .hdf5 для определенного сценария, который использует pytorch для классификации объектов.
Не было много информации о том, как использовать этот формат выходного файла в качестве модели в pytorch после поиска.
Что мне нужно сделать, чтобы использовать этот выходной файл в качестве обученной моделииспользовать в факеле? Буду признателен, если кто-нибудь укажет мне правильное направление, чтобы начать