diff --git a/tests/test_handler_transform_inverter.py b/tests/test_handler_transform_inverter.py
index ac369f068b..7be63e7638 100644
--- a/tests/test_handler_transform_inverter.py
+++ b/tests/test_handler_transform_inverter.py
@@ -107,7 +107,10 @@ def _train_func(engine, batch):
         original_name = data[-1]["label"]
         self.assertEqual(reverted_name, original_name)
         print("invert diff", reverted.size - n_good)
-        self.assertTrue((reverted.size - n_good) in (25300, 1812), "diff. in two possible values")
+        # 25300: 2 workers (cpu, non-macos)
+        # 1812: 0 workers (gpu or macos)
+        # 1824: torch 1.5.1
+        self.assertTrue((reverted.size - n_good) in (25300, 1812, 1824), "diff. in 3 possible values")
 
 
 if __name__ == "__main__":