Installation

From PyPI

pip install ane-trainer

From Source

git clone https://github.com/Lumi-node/ane-trainer.git
cd ane-trainer
pip install -e ".[dev]"

Requirements

• Python >= 3.10
• Dependencies: ane_trainer, numpy, platform, torch, torchvision