README.md

Title-Generation

2024 Applied Deep Learning Homework 2

Step 1: Install the evaluation metric
To install the evaluation metric, you can refer this repo: https://github.com/deankuo/ADL24-HW2

Download dataset and model checkpoint

To download the model checkpoint, you can run the commad:

bash ./download.sh

Reproducing best result

To reproduce best result, you can run the commad:

bash ./run.sh <data file> <output file>

Training

To train the model, you can run the commad:

bash ./train.sh

Reference

I have referenced the following repo, which is from huggingface.

Inference.py : https://github.com/huggingface/transformers/tree/main/examples/pytorch/summarization

Train.py : https://github.com/huggingface/transformers/blob/main/examples/pytorch/summarization/run_summarization.py