Wenn Sie den Artikel lesen und Vorschläge haben, können Sie sich gerne an uns wenden. Es wird ermutigend sein, wenn es Ihnen gefällt.
Wenn Sie Pytorch verwenden und den Wert eines Parameters einschränken möchten, kann dies mit der folgenden Methode realisiert werden.
for machen, ist die Verwendung von keys (), um auf alle Parameter der Reihe nach zuzugreifen und eine Funktion namens clamp anzuwenden. (Es ist np.clip in numpy.)state_dict = model.state_dict()#Parameter im Modell aufrufen
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_dicttorch.clamp begrenzen. Sie können den Wertebereich mit torch.clamp (Eingabe, min = 0, max = 10) begrenzen.Recommended Posts