Merge branch 'main' into feature/vqa

OFA-Sys · Apr 22, 2022 · 7f8c77b · 7f8c77b
2 parents d9bf799 + ff8f992
commit 7f8c77b
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/tasks/mm_tasks/refcoco.py b/tasks/mm_tasks/refcoco.py
@@ -38,6 +38,15 @@ class RefcocoConfig(OFAConfig):
     max_image_size: int = field(
         default=512, metadata={"help": "max image size for normalization"}
     )
+    scst: bool = field(
+        default=False, metadata={"help": "Self-critical sequence training"}
+    )
+    scst_args: str = field(
+        default='{}',
+        metadata={
+            "help": 'generation args for Self-critical sequence training, as JSON string'
+        },
+    )
 
 
 @register_task("refcoco", dataclass=RefcocoConfig)