#!/bin/bash
TIMM_TAG=$(curl -s https://api.github.com/repos/huggingface/pytorch-image-models/releases/latest | jq -r '.tag_name')

for file in train.py validate.py inference.py benchmark.py distributed_train.sh; do \
    curl -sLO "https://raw.githubusercontent.com/huggingface/pytorch-image-models/${TIMM_TAG}/${file}"; \
done

chmod +x distributed_train.sh