Izdan PyTorch 1.2.0

PyTorch, priljubljeno odprtokodno ogrodje za strojno učenje, je bilo posodobljeno na različico 1.2.0. Nova izdaja vključuje več kot 1900 popravkov, ki pokrivajo JIT, ONNX, načine porazdeljenega učenja in izboljšave zmogljivosti.

Nekaj ​​sprememb:

  • Nov API TorchScript omogoča nn.Module (vključno s podmoduli in metodami, klicanimi v forward()) je preprosto pretvoriti v ScriptModule.
  • Skupaj z Microsoftom je bila dodana popolna podpora za ONNX Opset različice 7 (v1.2), 8 (v1.3), 9 (v1.4) in 10 (v1.5). Poleg tega lahko uporabniki zdaj registrirajo lastne simbole za izvoz operacij po meri in med izvozom določijo dinamične velikosti vnosa.
  • podpora tensorboard ni več eksperimentalno.
  • Na podlagi članka dodan modul nn.Transformer Pozornost je vse, kar potrebujete.
  • Številne izboljšave API-ja C++.

Vir: linux.org.ru

Dodaj komentar