aboutsummaryrefslogtreecommitdiff
path: root/requirements.txt
diff options
context:
space:
mode:
authorgdamms <damguillotin@gmail.com>2024-07-02 11:39:08 +0200
committergdamms <damguillotin@gmail.com>2024-07-02 11:39:08 +0200
commit9d6d3f3c7f8f01e24d635c3f5d9b43fa697d5f9f (patch)
tree803c7a391acf67b2b4f37006a123af2f583f8ff3 /requirements.txt
parent965b433ac9e0b18c22cda8df1058876fe08dfb19 (diff)
downloaddiffusion-mnist-9d6d3f3c7f8f01e24d635c3f5d9b43fa697d5f9f.tar.gz
diffusion-mnist-9d6d3f3c7f8f01e24d635c3f5d9b43fa697d5f9f.zip
unteste update using troch-trainer
Diffstat (limited to 'requirements.txt')
-rw-r--r--requirements.txt3
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