Fonction torch.load_state_dict de PyTorch

Pytorch torch 参考手册Manuel de référence Pytorch torch

torch.load_state_dictIl s'agit d'une fonction dans PyTorch utilisée pour charger des dictionnaires d'état. Elle charge l'état des optimiseurs de modèle, etc., à partir d'un objet dictionnaire.

Définition de fonction

torch.load_state_dict(state_dict, strict=True)

Exemple d'utilisation

Exemple

import torch
import torch.nn as nn

# Définir un modèle simple
model = nn.Linear(10, 5)

# Enregistrer le dictionnaire d'état du modèle
state_dict = model.state_dict()
torch.save(state_dict, 'model_state.pt')

# Charger le dictionnaire d'état
loaded_state = torch.load('model_state.pt')
model.load_state_dict(loaded_state)
print("Dictionnaire d'état chargé avec succès !")

Pytorch torch 参考手册Manuel de référence Pytorch torch

Autres extensions