SeungoneKim / SICK_Summarization

[COLING 2022] Mind the Gap! Injecting Commonsense Knowledge for Abstractive Dialogue Summarization
25 stars 17 forks source link

AttributeError: 'DualDecoderTrainer' object has no attribute 'do_grad_scaling' #4

Open imanmorovatian opened 4 months ago

imanmorovatian commented 4 months ago

I wanted to train the model on samsum dataset with following command

!python3 src/train_summarization_full.py --finetune_weight_path="./weights_sickpp_samsum" --best_finetune_weight_path="./weights_sickpp_samsum_best" --dataset_name="samsum" --use_paracomet=True --model_name="facebook/bart-large-xsum" --epoch=10 --use_sentence_transformer True

but I got the following error

AttributeError: 'DualDecoderTrainer' object has no attribute 'do_grad_scaling'