Изменить гиперпараметр num_classes для SageMaker.

Question

Я выполняю дополнительное обучение на модели, которую я уже обучал в SageMaker. Я хочу добавить данные в существующие классы, а также создать новые классы. Первая модель имела 4 класса (num_classes = 4), но я хочу сохранить эти классы, а также добавить 3 дополнительных класса.

В документации сказано, что гиперпараметр num_classes должен быть одинаковым при выполнении пошагового обучения. Но если это так, это означает, что я не могу добавить классы в свою существующую модель, мне придется начинать с нуля каждый раз, когда я хочу изменить количество классов. Это точно? Или есть способ обновить существующую модель и изменить количество классов, на которых она обучается?

Вот пример записной книжки, которую я использую для дополнительной учебной работы: https://github.com/awslabs/amazon-sagemaker-examples/blob/master/introduction_to_amazon_algorithms/imageclassification_caltech/Image-classification-incremental-training-highlevel.ipynb

См. Параметр num_classes здесь

Answer 1

Для пошагового обучения со встроенным алгоритмом классификации изображений SageMaker входные гиперпараметры как новой модели, так и предварительно обученной модели должны иметь одинаковые настройки для входных параметров num_layers, image_shape и num_classes,Потому что эти параметры определяют сетевую архитектуру.

Как правило, для вновь добавленных классов вы создаете новый набор данных со старыми и новыми классами и обновляете веса моделей в новом наборе данных.Насколько я знаю, это невозможно с алгоритмом классификации изображений SageMaker.