Izlaista PyTorch 1.2.0

PyTorch, populārs atvērtā pirmkoda ietvars mašīnmācībai, ir atjaunināts uz versiju 1.2.0. Jaunajā laidienā ir iekļauti vairāk nekā 1900 labojumi, kas aptver JIT, ONNX, sadalītos mācību režīmus un veiktspējas uzlabojumus.

Dažas izmaiņas:

  • Jauna TorchScript API pieļauj Ir viegli konvertēt nn.Module (tostarp apakšmoduļus un metodes, kas tiek izsauktas forward()) par ScriptModule.
  • Kopā ar Microsoft ir pievienots pilns ONNX Opset versiju 7 (v1.2), 8 (v1.3), 9 (v1.4) un 10 (v1.5) atbalsts. Turklāt lietotāji tagad var reģistrēt savus simbolus pielāgoto operāciju eksportēšanai un eksportēšanas laikā norādīt dinamiskās ievades izmērus.
  • tensorboard atbalsts vairs nav eksperimentāls.
  • Pievienots nn.Transformatora modulis, pamatojoties uz rakstu Uzmanība ir viss, kas jums nepieciešams.
  • Daudzi C++ API uzlabojumi.

Avots: linux.org.ru

Pievieno komentāru