Сохранение веса одного слоя в Pytorch
Я хотел бы сохранить вес модели, но не всей модели:
torch.save(model, 'model.pth')
А точнее всего в один слой. например, предположим, я определил один слой следующим образом:
self.conv_up3 = convrelu(256 + 512, 512, 3, 1)
Как сохранить вес только этого слоя. А также как мне загрузить его для этого слоя.
Стоит ли изучать PHP в 2023-2024 годах?
Привет всем, сегодня я хочу высказать свои соображения по поводу вопроса, который я уже много раз получал в своем сообществе: "Стоит ли изучать PHP в...
Поведение ключевого слова "this" в стрелочной функции в сравнении с нормальной функцией
В JavaScript одним из самых запутанных понятий является поведение ключевого слова "this" в стрелочной и обычной функциях.
Приемы CSS-макетирования - floats и Flexbox
Здравствуйте, друзья-студенты! Готовы совершенствовать свои навыки веб-дизайна? Сегодня в нашем путешествии мы рассмотрим приемы CSS-верстки - в...
Тестирование функциональных ngrx-эффектов в Angular 16 с помощью Jest
В системе управления состояниями ngrx, совместимой с Angular 16, появились функциональные эффекты. Это здорово и делает код определенно легче для...
Концепция локализации и ее применение в приложениях React ⚡️
Локализация - это процесс адаптации приложения к различным языкам и культурным требованиям. Это позволяет пользователям получить опыт, соответствующий...
Пользовательский скаляр GraphQL
Листовые узлы системы типов GraphQL называются скалярами. Достигнув скалярного типа, невозможно спуститься дальше по иерархии типов. Скалярный тип...
0
0
11
1
Перейти к ответу
Данный вопрос помечен как решенный
Ответы 1
Ответ принят как подходящий
Вы можете сделать следующее, чтобы сохранить/получить параметры определенного слоя:
specific_params = self.conv_up3.state_dict()
# save/manipulate `specific_params` as you want
И аналогично, чтобы загрузить параметры для этого конкретного слоя:
self.conv_up3.load_state_dict(params)
Вы можете сделать это, потому что каждый слой сам по себе является нейронной сетью (экземпляром).
Другие вопросы по теме
У меня есть код, который я хочу перевести в pytorch. Я ищу способ перевести np.vectorize на любой способ pytorch в этом случае
Будут ли записываться градиенты при определении точности модели в pytorch?
Можно ли использовать `BatchNorm1d` PyTorch с `BCELossWithLogits`?
Использование целевого размера (torch.Size([16])), отличного от входного размера (torch.Size([16, 2])) устарело
Переупорядочить столбцы в тензоре по словарю
Numpy: фильтровать сегменты в массиве в зависимости от того, насколько они перекрываются со вторым массивом с другим типом сегментации
Длина набора данных изображения поезда и загрузчик изображения поезда отличаются?
Получить среднеквадратичную ошибку между каждым элементом в двух тензорах
Как типы данных, не являющиеся изображениями, такие как 3D-объекты, аудио, видео и т. д., работают с Activeloop Hub?
Как я могу отправить загрузчик данных на графический процессор в Google Colab?
Похожие вопросы
В Pytorch, как я могу перетасовать DataLoader?
Как изменить порядок тензора на основе тензора индексов того же размера
Факел model.load_state_dict *** AttributeError: объект «ModelName» не имеет атрибута «копия»
Для оптимизации цикла для создания матрицы смежности
Неверный синтаксис при копировании веса в pytorch
Какой элегантный способ сравнить результаты одноразовых этикеток?
Что делает Tensor[batch_mask, ...]?
Как узнать, что обученная модель верна?
Не могу импортировать torchvision
Я плохо расщепляюсь в pytorch