efficientnet_pytorch
PyYAML
torchaudio
einops