Installation

Requirements

 - python >= 3.8
 - pytorch >= 1.12.1  # For scatter_reduce
 - torchvision        # With matching version for your pytorch install
 - timm == 0.4.12     # Might work on other versions, but this is what we tested
 - jupyter            # For example notebooks
 - scipy              # For visualization and sometimes torchvision requires it

Setup

First, clone the repository:

git clone https://github.com/facebookresearch/tome
cd tome

Either install the requirements listed above manually, or use our conda environment:

conda env create --file environment.yml
conda activate tome

Then set up the tome package with:

python setup.py build develop