Add torch and torchvision dependencies, ensure venv is activated in script section

This commit is contained in:
Fam Zheng 2025-11-23 18:24:36 +00:00
parent 9d3a5053a3
commit ec65bfa1d2
2 changed files with 3 additions and 0 deletions

View File

@ -91,6 +91,7 @@ zy:
- source venv/bin/activate
- uv pip install -r requirements.txt --default-index https://pypi.tuna.tsinghua.edu.cn/simple
script:
- source venv/bin/activate
- make fetch-quick
- make sbs-quick
- make train-quick

View File

@ -19,6 +19,8 @@ matplotlib
scikit-learn
scipy
seaborn
torch
torchvision
kornia
opencv-python
opencv-contrib-python