Commit b444a9b
authored
Bump torch and torchvision to current versions (#25)
## Summary and motivation
The torch version here is >3 years out of date. This is what's
triggering the dependabot security alerts. Since we're still at the
start of the project, let's bump to the latest stable versions, so we
can 1) get the security fixes and 2) benefit from other improvements
(e.g. memory efficiency, more integrated CUDA support) over the past few
years.
## Testing
I ran `uv run python main train ...` locally simply to confirm that it
runs. Have not run it on a GPU yet. @hanaol would be good to test on
your cluster to make sure the large upgrade jump doesn't break anything.1 parent 3f3c677 commit b444a9b
1 file changed
+2
-2
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
1 | 1 | | |
2 | 2 | | |
3 | | - | |
4 | | - | |
| 3 | + | |
| 4 | + | |
5 | 5 | | |
6 | 6 | | |
7 | 7 | | |
| |||
0 commit comments