Skip to content

Project on Self-Supervised Learning for Prof. Yann LeCun's Deep Learning Course at NYU, Spring 2020

Notifications You must be signed in to change notification settings

MrinalJain17/ssl-project

Repository files navigation

Project on Self-Supervised Learning for Prof. Yann LeCun's Deep Learning Course at NYU, Spring 2020

  1. Model architecture and training files:

    python road_map_construction.py --EPOCHS 50 --BATCH_SIZE 128
    
  2. Sample notebook showing the generated roadmaps on validation data: Link

  3. The directory saved_models contains the trained model checkpoints.

  4. The code for object detection is inside the directory object_detection.

Requirements

  1. PyTorch
  2. Torchvision
  3. PyTorch Lightning - for simplifying the training, validation and testing interface.

About

Project on Self-Supervised Learning for Prof. Yann LeCun's Deep Learning Course at NYU, Spring 2020

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published