Add T5 Model Fine-Tuning Jupyter Notebook #71
Merged
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
This PR introduces a new Jupyter Notebook for fine-tuning the T5 model. The notebook includes steps for installing dependencies, loading and preprocessing the training data, training the T5 model, and saving the fine-tuned model. It also includes error catching and logging using the
loguru
library.The new notebook replaces the existing
fine_tune.ipynb
notebook and pulls in the existing training data within thedata/training
directory.Please review and let me know if any changes are required.
Thanks!
Fixes #67.
To checkout this PR branch, run the following command in your terminal: