Skip to content
/ CoCa Public

Capsule Network based Contrastive Learning of Unsupervised Visual Representations

Notifications You must be signed in to change notification settings

Harsh9524/CoCa

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

CoCa - Capsule Network based Contrastive Learning of Unsupervised Visual Representations

Python 3.8 harshpanwar

Official PyTorch Implementation of my MSc Thesis - Capsule Network based Contrastive Learning of Unsupervised Visual Representations.

Usage:

  1. Download the required modules using the command 'pip install -r requirements.txt'.
  2. Run the python file main.py to start training.
optional arguments:
--temperature                 Temperature used in softmax [default value is 0.2]
--k                           Top k most similar images used to predict the label [default value is 50]
--batch_size                  Number of images in each mini-batch [default value is 512]
--epochs                      Number of sweeps over the dataset to train [default value is 500]
--num_caps                    Number of capsules per layer [default value is 32]
--caps_size                   Number of neurons per capsule [default value is 64]
--depth                       Depth of additional layers [default value is 1]
--planes					  Starting layer width [default value is 16]
  1. Once the training is completed the trained weights and the training loss, top-1 and top-5 accuracy will be stored in a folder called 'results'.

About

Capsule Network based Contrastive Learning of Unsupervised Visual Representations

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages