Method __init__

codegeex/mindspore/src/sat_dataset.py:57–60

(self, dataset, seq_len, eod_id)

class PadDataset(Dataset):
    def __init__(self, dataset, seq_len, eod_id):
        self.dataset = dataset
        self.seq_len = seq_len + 1
        self.eod_id = eod_id

    def __len__(self):
        return len(self.dataset)