Fix test failures caused by D37088095 (part 2)

Summary:
Pull Request resolved: https://github.com/facebookresearch/d2go/pull/308

D37088095 made BC breaking changes in PyTorch, but missed
fixing a few callsites. This is the second diff to fix the affected
tests.

Reviewed By: jerryzh168, wat3rBro

Differential Revision: D37282853

fbshipit-source-id: 945c3628f53cefd2fbc2526630d1567a2b1cee25

d2go/modeling/meta_arch/rcnn.py

@@ -312,7 +312,7 @@ def default_rcnn_prepare_for_quant_convert(self, cfg):
         assert not isinstance(self.backbone, FPN), "FPN is not supported in FX mode"
         self.backbone = convert_fx(
             self.backbone,
-            convert_custom_config_dict={
+            convert_custom_config={
                 "preserved_attributes": ["size_divisibility", "padding_constraints"]
             },
         )

tests/runner/test_runner_lightning_task.py

@@ -186,7 +186,7 @@ class TestLightningTask(unittest.TestCase):
 
             def prepare_for_quant_convert(self, cfg):
                 self.avgpool = convert_fx(
-                    self.avgpool, convert_custom_config_dict=self.custom_config_dict
+                    self.avgpool, convert_custom_config=self.custom_config_dict
                 )
                 return self