機械学習用の人気のオープンソース フレームワークである 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