Si vous lisez l'article et avez des suggestions, n'hésitez pas à nous contacter. Ce sera encourageant si vous l'aimez.
Lorsque vous utilisez Pytorch et que vous souhaitez contraindre la valeur d'un paramètre, cela peut être réalisé par la méthode suivante.
min = 1e-4.
--Ce que nous faisons avec for, c'est utiliser keys () pour accéder à tous les paramètres dans l'ordre et appliquer une fonction appelée clamp. (C'est «np.clip» dans «numpy».)state_dict = model.state_dict()#Appel de paramètres dans le modèle
for k in state_dict.keys():
state_dict[k] = torch.clamp(state_dict[k], min=1e-4)
model.load_state_dict(state_dict)
torch.clampmodel.load_state_dict--Par exemple, si vous souhaitez limiter la valeur minimale à 0, vous pouvez utiliser la fonction torch.clamp pour limiter la valeur. Vous pouvez limiter la plage de valeurs avec torch.clamp (input, min = 0, max = 10).
model.load_state_dict.Recommended Posts