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.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
data
models
.gitignore
README.md
playground.ipynb

README.md

pytorch_models

Self-supervised Models

In models/self_supervised.py, 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

TODO-models:

  • weakly_colorization
  • weakly_splitbrain

TODO-misc:

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

Model definition and weights were taken from http://netdissect.csail.mit.edu.

You can’t perform that action at this time.