Skip to content
No description, website, or topics provided.
Jupyter Notebook Python
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Type Name Latest commit message Commit time
Failed to load latest commit information.


Self-supervised Models

In models/, we provide a stand-alone file that provides model definitions for self-supervised models converted from Caffe to PyTorch.

model = models.self_supervised.alexnet(task_name='weakly_solvingpuzzle', pretrained=True)

The following models have been converted and fully verified to be equivalent:

  • weakly_deepcontext
  • weakly_learningbymoving
  • weakly_objectcentric
  • weakly_solvingpuzzle
  • weakly_videoorder
  • weakly_videotracking

These models have been verified to be equivalent in the forward pass but have discrepancies in the backwards pass (i.e., gradients):

  • weakly_audio
  • weakly_egomotion


  • weakly_colorization
  • weakly_splitbrain


  • define get_transform to provide the right data transformations for each task.

Model definition and weights were taken from

You can’t perform that action at this time.