This repository contains the official PyTorch implementation for MoE-FFD.
- python == 3.8.5
- timm == 0.5.4
- pytorch == 1.8.0
- albumentations == 1.1.0
To train the model, use the train command as follow:
python train.py
To evaluate the trained model, use the evaluation command as follow:
python eval.py --model_path {your trained model}