
[pytorch]
torch==1.10.1
torchvision==0.11.2
torchaudio==0.10.1
pytorch-lightning==1.6.5
simplejpeg
albumentations

[tensorflow]
tensorflow>=2.10.0
protobuf<=3.19
