Programmiererforum

Code: Select all

class checkpoint_t(NamedTuple):
epoch: int
model_state_dict: Dict[str, Any]
optimizer_state_dict: Dict[str, Any]
model_name: str | None = None

< /code>
Wie auch immer ich nach dem Speichern diesen benannten Tuple über < /p>
nicht laden kannimport torch
from train import checkpoint_t
with torch.serialization.safe_globals([checkpoint_t]):
print("safe globals: ", torch.serialization.get_safe_globals())
checkpoint: checkpoint_t = torch.load(parsed_args.checkpoint, weights_only=True)
< /code>
Es wird immer noch gesagt: < /p>

WOHNUNGSUMPICKLER -ERROR: Nicht unterstütztes Global: Global  Main < /strong> .Cockpoint_t war standardmäßig keine erlaubte globale. Bitte verwenden Sie t Torch.Serialization.add_Safe_Globals ([checkpoint_t])