accelerate>=0.12.0
evaluate>=0.2.2
transformers>=4.21.1
timm>=0.6.7
click==8.0.4
pytorchvideo
torch
torchvision
balanced-loss
scikit-learn
tensorboard
opencv-python

[dev]
black==21.7b0
flake8==3.9.2
isort==5.9.2

[test]
black==21.7b0
flake8==3.9.2
isort==5.9.2
onnx
onnxruntime
