transformer-pipeline pip3 install torch torchvision transformers[torch] albumentations datasets pycocotools timm gitpython Training only pip install torchmetrics