- Clone the Repo
git clone https://github.com/iitaakash/cs6910_assignment3.git
- To train the model run. Edit the line in
# define the model here
model = AttentionModel()
# model = NormalModel()
train.py
to switch between model with attention and the default model(as shown above) and then run
python train.py
- For testing
python test.py
Edit the model path in test.py
as shown below
model = torch.load("best_model.pth")