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, основан на статията Attention Is All You Need.
- Множество подобрения в C++ API.
Източник: linux.org.ru
