Iznācis PyTorch 1.2.0

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

Daži jaunumi:

• Jaunais TorchScript API ļauj vieglāk konvertēt nn.Module (ieskaitot apakšmoduļus un metodes, kas izsauktas forward()) uz ScriptModule.
• Sadarbībā ar Microsoft ir pievienota pilna atbalsts ONNX Opset versijām 7(v1.2), 8(v1.3), 9(v1.4) un 10 (v1.5). Turklāt lietotāji tagad var reģistrēt savus simbolus, lai eksportētu pielāgotas operācijas un norādīt dinamiskos ievades datu izmērus eksporta laikā.
• Atbalsts tensorboard vairs nav eksperimentāls.
• Pievienots modulis nn.Transformer, balstoties uz rakstu Attention Is All You Need.
• Daudz uzlabojumu C++ API.

Avots: linux.org.ru