einops mlflow opencv-python pytorch-lightning==1.1.4 torchvision==0.2.2.post3 tqdm yacs protobuf==3.19.6