This repository contains a Transformer-based translation model implemented in PyTorch. The model is designed to translate text from French to English using a sequence-to-sequence architecture with attention mechanisms.
This Notebook was developed during the course Natual Language Processing at the University of Groningen. This project was assesed with a 95/100.
- Python 3.9
- PyTorch 1.8.0
- Transformers 4.27.0
- Datasets 2.10.0
- NLTK 3.8.1
- NumPy <2
- ipywidgets
- Matplotlib
-
Clone this repository:
git clone <repository-url> cd <repository-directory>
-
Create a virtual environment (optional but recommended):
python -m venv venv source venv/bin/activate # On Windows: venv\Scripts\activate
-
Install dependencies:
pip install -r requirements.txt
-
For GPU support (optional):
pip install torch==1.8.0+cu111 -f https://download.pytorch.org/whl/torch_stable.html