PyTorch torch.load_state_dict función
Manual de referencia de Pytorch torch
torch.load_state_dictEs una función en PyTorch utilizada para cargar el diccionario de estado. Carga el estado del modelo, optimizador, etc., desde el objeto de diccionario.
Definición de la función
torch.load_state_dict(state_dict, strict=True)
Ejemplo de uso
Ejemplo
import torch
import torch.nn as nn
# Definir un modelo simple
model = nn.Linear(10, 5)
# Guardar el diccionario de estado del modelo
state_dict = model.state_dict()
torch.save(state_dict, 'model_state.pt')
# Cargar el diccionario de estado
loaded_state = torch.load('model_state.pt')
model.load_state_dict(loaded_state)
print("State dict loaded successfully!")
import torch.nn as nn
# Definir un modelo simple
model = nn.Linear(10, 5)
# Guardar el diccionario de estado del modelo
state_dict = model.state_dict()
torch.save(state_dict, 'model_state.pt')
# Cargar el diccionario de estado
loaded_state = torch.load('model_state.pt')
model.load_state_dict(loaded_state)
print("State dict loaded successfully!")
Otras extensiones