[ci] Re-run integration ground truth from fairseq

Adopted best practice set by @patrickvonplaten of commenting lines run on fairseq, for easy comparison

also see #3020
This commit is contained in:
Julien Chaumond
2020-03-03 15:31:31 -05:00
parent 5b396457e5
commit f631e01d2c
4 changed files with 39 additions and 24 deletions

View File

@@ -46,7 +46,9 @@ logger = logging.getLogger(__name__)
SAMPLE_TEXT = "Hello world! cécé herlolip"
def convert_roberta_checkpoint_to_pytorch(roberta_checkpoint_path, pytorch_dump_folder_path, classification_head):
def convert_roberta_checkpoint_to_pytorch(
roberta_checkpoint_path: str, pytorch_dump_folder_path: str, classification_head: bool
):
"""
Copy/paste/tweak roberta's weights to our BERT structure.
"""