diff options
Diffstat (limited to 'requirements.txt')
| -rw-r--r-- | requirements.txt | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/requirements.txt b/requirements.txt index dce3faf..d590e96 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,5 @@ torchvision==0.15.2 matplotlib==3.8.0 rich==13.3.5 -PyQt5==5.15.10
\ No newline at end of file +PyQt5==5.15.10 +torch-trainer @ git+https://github.com/gdamms/torch-trainer.git@da70ca78adf6195f2e3add92938b9c282d5c94cb
\ No newline at end of file |