Skip to content

Latest commit

 

History

History
25 lines (16 loc) · 782 Bytes

README.md

File metadata and controls

25 lines (16 loc) · 782 Bytes

Reproduction of Cross-Encoder for Unsupervised Gaze Representation Learning

This repository contains the code for a PyTorch reproduction of the Cross-Encoder from the paper Cross-Encoder for "Unsupervised Gaze Representation Learning", by Sun et al.

Read the blog post for more information.

Authors

Installation

To install this project's dependencies run

  pip install -r requirements.txt

Then, save the Columbia Gaze dataset to a data folder. To preprocess the dataset and start training the Cross-Encoder run

  python train_cross_encoder.py