Izlaista PyTorch 1.5.0

PyTorch, populārs mašīnmācīšanās ietvars, ir atjaunināts uz versiju 1.5.0. Šajā laidienā ir iekļauti vairāki būtiski API papildinājumi un uzlabojumi, tostarp:

  • C++ API, kas iepriekš tika uzskatīta par eksperimentālu, beidzot ir stabilizēta. Lietotāji tagad var viegli pārtulkot savus modeļus no Python API uz C++ API.

  • Torch.distributed.rpc pakotne ir stabilizēta, nodrošinot plašas iespējas sadalītā apmācībā, tostarp automātisku gradientu aprēķinu un modeļa parametru atjaunināšanu.

  • Atjaunināts torch_xla — pakotne, kas izmanto XLA kompilatoru, lai paātrinātu apmācību modeļus mākoņa TPU.

  • Ir atjauninātas arī torcaudio, torchvision un torchtext pakotnes, nodrošinot rīkus tādu modeļu izstrādei, kas apstrādā audio, grafiskos un teksta datus.

  • Python 2 vairs netiek atbalstīts. Visa turpmākā attīstība tiks veikta tikai Python 3.

Avots: linux.org.ru

Pievieno komentāru