PyTorch 1.2.0 julkaistu

PyTorch, suosittu avoimen lähdekoodin kehys koneoppimiseen, on päivitetty versioon 1.2.0. Uusi julkaisu sisältää yli 1900 XNUMX korjausta, jotka kattavat JIT, ONNX, hajautetut oppimistilat ja suorituskyvyn parannukset.

Joitain muutoksia:

  • Uusi TorchScript API sen avulla Se on helppo muuntaa nn.Module (mukaan lukien alimoduulit ja menetelmät, joita kutsutaan eteenpäin()) ScriptModuleksi.
  • Yhdessä Microsoftin kanssa on lisätty täysi tuki ONNX Opset -versioille 7 (v1.2), 8 (v1.3), 9 (v1.4) ja 10 (v1.5). Lisäksi käyttäjät voivat nyt rekisteröidä omia symbolejaan mukautettujen toimintojen vientiä varten ja määrittää dynaamisia syöttökokoja viennin aikana.
  • tensorilevyn tuki ei ole enää kokeellinen.
  • Lisätty nn.Transformer-moduuli artikkelin perusteella Huomio on kaikki mitä tarvitset.
  • Lukuisia parannuksia C++ API:hen.

Lähde: linux.org.ru

Lisää kommentti