Trained model weights

Checkpoint .pth files of the models that were trained on the dataset, to run with test.py