PyTorch 1.2.0 шығарылды

PyTorch, машиналық оқытуға арналған танымал ашық бастапқы коды 1.2.0 нұсқасына жаңартылды. Жаңа шығарылым JIT, ONNX, бөлінген оқыту режимдері мен өнімділікті жақсартуды қамтитын 1900-ден астам түзетулерді қамтиды.

Кейбір өзгерістер:

  • Жаңа TorchScript API ол мүмкіндік береді nn.Module (ішкі модульдер мен forward() деп аталатын әдістерді қоса) ScriptModule-ге түрлендіру оңай.
  • Microsoft корпорациясымен бірге ONNX Opset 7 (v1.2), 8 (v1.3), 9 (v1.4) және 10 (v1.5) нұсқаларына толық қолдау қосылды. Сонымен қатар, пайдаланушылар енді теңшелетін операция экспорттары үшін өздерінің таңбаларын тіркей алады және экспорттау кезінде динамикалық енгізу өлшемдерін көрсете алады.
  • тензорбордты қолдау енді жоқ эксперименттік.
  • Мақалаға негізделген nn.Transformer модулі қосылды Сізге тек назар аудару қажет.
  • C++ API үшін көптеген жақсартулар.

Ақпарат көзі: linux.org.ru

пікір қалдыру