中文:
运行微调程序,在
Trainer(model=model, args=training_args, train_dataset=train_dataset,
eval_dataset=val_dataset, data_collator=lambda data: {'input_ids': torch.stack([f[0] for f in data]),
'attention_mask': torch.stack([f[1] for f in data]),
'labels': torch.stack([f[0] for f in data])}).train()
行报错,报错信息如下:
0%| | 0/7926 [00:00<?, ?it/s]
Killed
请问要如何解决?
English:
Run the fine tuning program and report an error on the
Trainer(model=model, args=training_args, train_dataset=train_dataset,
eval_dataset=val_dataset, data_collator=lambda data: {'input_ids': torch.stack([f[0] for f in data]),
'attention_mask': torch.stack([f[1] for f in data]),
'labels': torch.stack([f[0] for f in data])}).train()
line, the error message is as follows:
0%| | 0/7926 [00:00<?, ?it/s]
Killed
How to solve it?