diff --git a/requirements.txt b/requirements.txt index a908358..64466ed 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,4 @@ -torch==1.4.0 +torch==2.2.0 opencv==4.2 pthflops torchsummary \ No newline at end of file