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

This commit is contained in:
Julien Chaumond committed 2020-03-24 17:56:40 -04:00
1 parent f8823bad9a
commit eaabaaf750
1 file changed
+1 -1
+1 -1
View File
@@ -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)