MoE-FFD

This repository contains the official PyTorch implementation for MoE-FFD.

Requirements

• python == 3.8.5
• timm == 0.5.4
• pytorch == 1.8.0
• albumentations == 1.1.0

Training:

To train the model, use the train command as follow:

python train.py

Evaluate:

To evaluate the trained model, use the evaluation command as follow:

python eval.py --model_path {your trained model}