PyTorch 1.2.0 đã được phát hành

PyTorch, một framework nguồn mở phổ biến dành cho machine learning, đã được cập nhật lên phiên bản 1.2.0. Bản phát hành mới bao gồm hơn 1900 bản sửa lỗi bao gồm JIT, ONNX, chế độ học tập phân tán và cải tiến hiệu suất.

Một vài thay đổi:

  • API TorchScript mới cho phép Thật dễ dàng để chuyển đổi nn.Module (bao gồm các mô-đun con và các phương thức được gọi trong Forward()) thành ScriptModule.
  • Cùng với Microsoft, hỗ trợ đầy đủ cho ONNX Opset phiên bản 7 (v1.2), 8 (v1.3), 9 (v1.4) và 10 (v1.5) đã được thêm vào. Ngoài ra, giờ đây người dùng có thể đăng ký các ký hiệu của riêng mình để xuất hoạt động tùy chỉnh và chỉ định kích thước đầu vào động trong quá trình xuất.
  • hỗ trợ tensorboard không còn nữa thực nghiệm.
  • Đã thêm mô-đun nn.Transformer dựa trên bài viết Chú ý là tất cả những gì bạn cần.
  • Nhiều cải tiến cho API C++.

Nguồn: linux.org.ru

Thêm một lời nhận xét