Bug fix

pytorch · Jun 19, 2024 · 7c98b1c · 7c98b1c
1 parent ec9fcc6
commit 7c98b1c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchbenchmark/models/pytorch_unet/install.py b/torchbenchmark/models/pytorch_unet/install.py
@@ -1,4 +1,4 @@
 from utils.python_utils import pip_install_requirements
 
 if __name__ == '__main__':
-    pip_install_requirements()
+    pip_install_requirements(requirements_txt="pytorch_unet/requirements.txt")