/ContrastiveLearningSM

Contrastive Learning with Task Similarity Matrix

Primary LanguagePureBasic

ContrastiveLearningTSM

Contrastive Learning with Task Similarity Matrix

  1. Change model_path = "models/unsupervised_combined" in train.py to the appropriate model
  2. Simply run python3 -m train to check the model's performance over the evaluation data-set.