[run_language_modeling] Fix: initialize a new model from a config object

2020-03-24 17:56:40 -04:00
parent f8823bad9a
commit eaabaaf750
1 changed files with 1 additions and 1 deletions
--- a/examples/run_language_modeling.py
+++ b/examples/run_language_modeling.py
@@ -710,7 +710,7 @@ def main():
        )
    else:
        logger.info("Training new model from scratch")
-        model = AutoModelWithLMHead(config=config)
+        model = AutoModelWithLMHead.from_config(config)

    model.to(args.device)