Выйшаў PyTorch 1.2.0

PyTorch - папулярны адкрыты фрэймворк для машыннага навучання абнавіўся да версіі 1.2.0. Новы рэліз уключае ў сябе больш за 1900 выпраўленняў, якія ахопліваюць JIT, ONNX, размеркаваныя рэжымы навучання і паляпшэння прадукцыйнасці.

Некаторыя змены:

  • Новы TorchScript API дазваляе лёгка сканвертаваць nn.Module (уключаючы падмодулі і метады, якія выклікаюцца ў forward()) у ScriptModule.
  • Сумесна з Microsoft дададзена поўная падтрымка ONNX Opset версій 7 (v1.2), 8 (v1.3), 9 (v1.4) і 10 (v1.5). Акрамя таго, зараз карыстачы могуць зарэгістраваць свае ўласныя знакі для экспарту кастамных аперацый і паказваць дынамічныя памеры ўваходных дадзеных падчас экспарту.
  • Падтрымка tensorboard больш не з'яўляецца эксперыментальнай.
  • Дададзены модуль nn.Transformer, заснаваны на артыкуле Увага - усё, што вам трэба.
  • Шматлікія паляпшэнні ў C++ API.

Крыніца: linux.org.ru

Дадаць каментар