PyTorch 1.2.0 udgivet

PyTorch, en populær open source-ramme til maskinlæring, er blevet opdateret til version 1.2.0. Den nye udgivelse indeholder mere end 1900 rettelser, der dækker JIT, ONNX, distribuerede læringstilstande og ydeevneforbedringer.

Nogle ændringer:

  • Ny TorchScript API Det gør det muligt Det er nemt at konvertere nn.Module (inklusive undermoduler og metoder kaldet i forward()) til ScriptModule.
  • Sammen med Microsoft er fuld understøttelse af ONNX Opset version 7 (v1.2), 8 (v1.3), 9 (v1.4) og 10 (v1.5) blevet tilføjet. Derudover kan brugere nu registrere deres egne symboler til brugerdefinerede operationseksporter og angive dynamiske inputstørrelser under eksport.
  • tensorboard understøttelse er ikke længere eksperimentel.
  • Tilføjet nn.Transformer-modul baseret på artiklen Opmærksomhed er alt hvad du behøver.
  • Talrige forbedringer til C++ API.

Kilde: linux.org.ru

Tilføj en kommentar