From 0a83037ef0c592124cd18d71189c5acb49dc661f Mon Sep 17 00:00:00 2001 From: system Date: Fri, 31 Jan 2020 23:00:25 +0000 Subject: [PATCH] Update config.json --- config.json | 3 +++ 1 file changed, 3 insertions(+) diff --git a/config.json b/config.json index b26f520..565bc76 100644 --- a/config.json +++ b/config.json @@ -1,4 +1,7 @@ { + "architectures": [ + "GPT2LMHeadModel" + ], "attn_pdrop": 0.1, "embd_pdrop": 0.1, "initializer_range": 0.02,