| set -e | |
| echo "=== job start: $(date) ===" | |
| nvidia-smi || true | |
| pip install -q --no-cache-dir -r /code/code/requirements.txt | |
| echo "=== deps installed: $(date) ===" | |
| python -c "import torch, torchvision, transformers, datasets, accelerate, pycocotools, huggingface_hub; print('torch', torch.__version__, 'cuda', torch.cuda.is_available(), 'tf', torchvision.__version__, 'transformers', transformers.__version__, 'datasets', datasets.__version__)" | |
| exec python -u /code/code/train_detr.py "$@" | |