gustavla / self-supervision

Training and evaluating self-supervised deep neural networks
BSD 3-Clause "New" or "Revised" License
29 stars 6 forks source link

Self-supervision

Training and evaluating self-supervised deep neural networks.

Models

These models were all trained entirely without labeled data using colorization as a proxy task:

The models are saved in Caffe HDF5 files. Our evaluation code in TensorFlow can load these files. The AlexNet model uses LRN (applied before pooling)

Paper