Taschenlampe kann Safe_Globals nicht namensTuple -Klasse hinzufügen ⇐ Python

[Anonymous]

Ich definiere ein paar namens Tuple wie folgt: < /p>

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 -Fehler: Nicht unterstütztes global: global __main __. Checkpoint_t < /code> war kein standardmäßig global erlaubt. Bitte verwenden Sie t Torch.Serialization.add_Safe_Globals ([checkpoint_t]) 

oder theSerialization.Safe_Globals ([Checkpoint_t]) Context Manager, um diese globale zu ermöglichen, wenn Sie dieser Klasse/Funktion vertrauen.
< /blockquote>
Irgendeine Idee warum? und wie kann ich das beheben?