martin6336 / dsgsum Goto Github PK

The data has been preprocessed, run train.py, this error occurs：
Traceback (most recent call last):
File "/home/mist/src/train.py", line 140, in
train_abs(args, device_id)
File "/home/mist/src/train_abstractive.py", line 276, in train_abs
train_abs_single(args, device_id)
File "/home/mist/src/train_abstractive.py", line 339, in train_abs_single
trainer.train(train_iter_fct, args.train_steps, device_id)
File "/home/mist/src/models/trainer.py", line 145, in train
for i, batch in enumerate(train_iter):
File "/home/mist/src/models/data_loader.py", line 205, in iter
for batch in self.cur_iter:
File "/home/mist/src/models/data_loader.py", line 363, in iter
for idx, minibatch in enumerate(self.batches):
File "/home/mist/src/models/data_loader.py", line 343, in create_batches
for buffer in self.batch_buffer(data, self.batch_size * 300):
File "/home/mist/src/models/data_loader.py", line 311, in batch_buffer
ex = self.preprocess(ex, self.is_test)
File "/home/mist/src/models/data_loader.py", line 256, in preprocess
tgt = ex['tgt'][:self.args.max_tgt_len][:-1]+[2]
KeyError: 'tgt'

Do not reproduce the thesis, there is no prepro processing file,
good jobs,bro