Add torch and torchvision dependencies, ensure venv is activated in script section
This commit is contained in:
parent
9d3a5053a3
commit
ec65bfa1d2
@ -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
|
||||
@ -19,6 +19,8 @@ matplotlib
|
||||
scikit-learn
|
||||
scipy
|
||||
seaborn
|
||||
torch
|
||||
torchvision
|
||||
kornia
|
||||
opencv-python
|
||||
opencv-contrib-python
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user