https://github.com/cyrusvahidi/torch-lightning-triplet

Online Triplet Loss - Pytorch Lightning - MNIST
https://github.com/cyrusvahidi/torch-lightning-triplet

Last synced: 2 months ago
JSON representation

Online Triplet Loss - Pytorch Lightning - MNIST

• Host: GitHub
• URL: https://github.com/cyrusvahidi/torch-lightning-triplet
• Owner: cyrusvahidi
• Created: 2021-01-17T16:29:32.000Z (over 4 years ago)
• Default Branch: main
• Last Pushed: 2021-01-17T16:40:33.000Z (over 4 years ago)
• Last Synced: 2025-02-08T13:42:50.000Z (4 months ago)
• Language: Python
• Size: 11.7 KB
• Stars: 2
• Watchers: 2
• Forks: 0
• Open Issues: 1
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
# torch-lightning-triplet

* Triplet Loss implemented with PyTorch Lightning 

## Installation

`git clone`

#### Create a virtual environment

`python3 -m venv `

`source venv/bin/activate`

#### Install the dependencies

`cd torch-lightning-triplet`

`pip install -r requirements.txt`

## Train the model

`python -m src.scripts.train_triplet` 

see command-line arguments in `src.utils.parse_args`

* Embeddings visualisations on the test set before and after training are output to `/plot`

## TODO

- [ ] implement hard triplet loss