Загрузка частичных весов нейронной сети

Я работаю над проектом глубокого обучения, в котором я добавляю к предварительно обученной Wide ResNet пару линейных слоев после этого и еще одну технику, которую я использую.

Мой вопрос в том, если я создаю сеть, в которой я добавляю слои и техники, которые я использую, к Wide ResNet как ModuleList, а затем загружаю предварительно обученные веса для всей модели с дополнениями, вызывает ли это ошибку в процедуре загрузки?

или она знает, что должна загружать веса только частично (т. е. только при сбросе).

Пример codeа:

или он должен выглядеть примерно так?

Другими словами, как загрузка весов в PyTorch работает с помощью torch.load?

Спасибо

model = get_model(backbone) # Load ResNet
model = ConstructNewModel(model) # Add layers
load_weights(model, pretrained_path) # load ResNet Weights
model = model.to("cuda") # Add to GPU

model = get_model(backbone) # Load ResNet
load_weights(model, pretrained_path) # load ResNet Weights
model = ConstructNewModel(model) # Add layers
model = model.to("cuda") # Add to GPU