PyTorch torch.load_state_dict-Funktion
PyTorch torch Referenzhandbuch
torch.load_state_dictEs ist eine Funktion in PyTorch, die zum Laden von State-Dictionaries verwendet wird. Sie lädt den Zustand von Modelloptimierern usw. aus einem Dictionary-Objekt.
Funktionsdefinition
torch.load_state_dict(state_dict, strict=True)
Anwendungsbeispiel
Beispiel
import torch
import torch.nn as nn
# Ein einfaches Modell definieren
model = nn.Linear(10, 5)
# Modell-State-Dictionary speichern
state_dict = model.state_dict()
torch.save(state_dict, 'model_state.pt')
# State-Dictionary laden
loaded_state = torch.load('model_state.pt')
model.load_state_dict(loaded_state)
print("State-Dictionary erfolgreich geladen!")
import torch.nn as nn
# Ein einfaches Modell definieren
model = nn.Linear(10, 5)
# Modell-State-Dictionary speichern
state_dict = model.state_dict()
torch.save(state_dict, 'model_state.pt')
# State-Dictionary laden
loaded_state = torch.load('model_state.pt')
model.load_state_dict(loaded_state)
print("State-Dictionary erfolgreich geladen!")
Andere Erweiterungen