albumentations evaluate opencv-python optuna pytorch-lightning torch transformers