Update requirements version for PT 3dunet (#1382)

add version limit
intel · Nov 13, 2023 · f5167dc · f5167dc
1 parent fa929b6
commit f5167dc
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/examples/pytorch/image_recognition/3d-unet/quantization/ptq/fx/requirements.txt b/examples/pytorch/image_recognition/3d-unet/quantization/ptq/fx/requirements.txt
@@ -1,6 +1,8 @@
 neural-compressor
 numpy<=1.23.5
-batchgenerators==0.20.0
+nibabel==5.0.0
+SimpleITK==2.2.1
+batchgenerators==0.21.0
 --find-links https://download.pytorch.org/whl/torch_stable.html
 torch>=1.6.0+cpu
 torchvision>=0.7.0+cpu