Κυκλοφόρησε το PyTorch 1.2.0

Το PyTorch, ένα δημοφιλές πλαίσιο ανοιχτού κώδικα για μηχανική μάθηση, έχει ενημερωθεί στην έκδοση 1.2.0. Η νέα έκδοση περιλαμβάνει περισσότερες από 1900 επιδιορθώσεις που καλύπτουν JIT, ONNX, κατανεμημένες λειτουργίες εκμάθησης και βελτιώσεις απόδοσης.

Μερικές αλλαγές:

  • Νέο TorchScript API позволяет Είναι εύκολο να μετατρέψετε το nn.Module (συμπεριλαμβανομένων υπομονάδων και μεθόδων που καλούνται στο forward()) σε ScriptModule.
  • Μαζί με τη Microsoft, προστέθηκε πλήρης υποστήριξη για τις εκδόσεις 7 (v1.2), 8 (v1.3), 9 (v1.4) και 10 (v1.5) του ONNX Opset. Επιπλέον, οι χρήστες μπορούν πλέον να καταχωρούν τα δικά τους σύμβολα για εξαγωγές προσαρμοσμένων λειτουργιών και να καθορίζουν δυναμικά μεγέθη εισόδου κατά την εξαγωγή.
  • υποστήριξη τανσοπλακ δεν είναι πλέον πειραματικός.
  • Προστέθηκε μονάδα nn.Transformer με βάση το άρθρο Η προσοχή είναι το μόνο που χρειάζεστε.
  • Πολυάριθμες βελτιώσεις στο C++ API.

Πηγή: linux.org.ru

Προσθέστε ένα σχόλιο