x/themblem
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
.gitlab-ci.yml
View File

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

2
requirements.txt
View File

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