typo

mpelchat04 · Oct 4, 2024 · 23a8589 · 23a8589
1 parent ccba099
commit 23a8589
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/inference_segmentation.py b/inference_segmentation.py
@@ -108,7 +108,7 @@ def main(params:Union[DictConfig, Dict]):
                                  device=device_str,
                                  gpu_id=gpu_index,
                                  prediction_threshold=prediction_threshold,
-                                 transformer=True,
+                                 transformers=True,
                                  transformer_flip=transform_flip,
                                  transformer_rotate=transform_rotate,
                                  )