search

THUNLP-MT / THUMT

An open-source neural machine translation toolkit developed by Tsinghua Natural Language Processing Group
BSD 3-Clause "New" or "Revised" License
701 stars 197 forks source link

Fix the lacking abstract method in dataset and typing error since pytorch 1.9 #107

Closed aseaday closed 2 years ago

aseaday commented 2 years ago
  1. Fix the lacking abstract method in the subclass of Dataset.
  2. Since pytorch 1.9, class torch.data.iterableDataset 'siter method must be annotated as typing.Iterator or collections.abc.iterator