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, заснований на статті Увага – це все, що вам потрібно.
- Численні поліпшення C++ API.
Джерело: linux.org.ru