enhanced_transfer_learning / environment.yml
Andrzej Daniel Dobrzycki
First version
89b1f42
raw
history blame
590 Bytes
name: enhanced-tl
dependencies:
- pip
- pip:
- torch==2.0.1
- torchvision==0.15.2
- onnxruntime==1.15.1
- requests==2.32.3
- PyYAML==6.0.1
- scipy==1.13.0
- py-cpuinfo==9.0.0
- gradio==4.31.5
- onnxslim==0.1.31
- ultralytics==8.2
- onnx==1.14.0
- opencv-python==4.9.0.80
- wandb==0.15.7
- onnx==1.14.0
- onnxruntime-gpu==1.18.0
- huggingface-hub==0.23.2
- pycocotools==2.0.7
- safetensors==0.4.3
- psutil==5.9.8
- torch==2.0.1
- wget==3.2
- ttach==0.0.3