Kaydetme ve yükleme
StateDict, NXM1 biçimi ve checkpoint sınırları.
01StateDict anlık görüntüsü#
state_dict parametre verilerini ve buffer’ları clone eder. Anahtarlar 0.weight veya norm.running_mean gibi hiyerarşiktir. Gradyan, optimizer, RNG, eğitim modu ve mimari saklanmaz. Yüklemeden önce aynı mimariyi kurun.
save_state_dict(model.state_dict(), "model.nxm");auto state=load_state_dict_file("model.nxm");model.load_state_dict(state); // strict=truemodel.eval();02Strict yükleme#
load_state_dict(state,strict=true) eksik/fazla anahtarları reddeder. strict=false onları tolere eder; eşleşen anahtarların şekilleri yine uymalıdır. Değerler mevcut parametrelere kopyalanır.
03NXM1 düzeni#
NXM1 sihri + uint32 kayıt sayısı. Her kayıt: uint32 isim uzunluğu, isim baytları, uint32 rank, uint64 boyutlar, ham float32. Host byte sırası kullanılır; endian dönüşümü yoktur.
04Doğrulama sınırları#
Okuyucu sihir, rank≤8 ve kesilmiş dosya kontrolü yapar; toplam boyut, isim uzunluğu veya kayıt sayısı kotası yoktur. Kontrollü checkpoint alışverişi içindir. Güvenilmeyen yüklemeler öncesinde sınırlandırılmış doğrulama ekleyin.