From eaabaaf750613e00fdece7245ef04ca3d1a775a9 Mon Sep 17 00:00:00 2001 From: Julien Chaumond Date: Tue, 24 Mar 2020 17:56:40 -0400 Subject: [PATCH] [run_language_modeling] Fix: initialize a new model from a config object --- examples/run_language_modeling.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/run_language_modeling.py b/examples/run_language_modeling.py index 8e8694533c..5ab9ca31f9 100644 --- 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)