embeddings resizing + tie_weights
This commit is contained in:
1 parent
50e62a4cb4
commit
bd404735a7
15 files changed
+196
-332
No files matched your search
@@ -28,7 +28,7 @@ import torch
|
||||
from pytorch_transformers import (GPT2Config, GPT2Model,
|
||||
GPT2LMHeadModel, GPT2DoubleHeadsModel)
|
||||
|
||||
from .modeling_tests_commons import (create_and_check_commons, ConfigTester, GPTModelTester)
|
||||
from .modeling_common_test import (create_and_check_commons, ConfigTester, GPTModelTester)
|
||||
|
||||
class GPT2ModelTest(unittest.TestCase):
|
||||
|
||||
|
||||
Reference in new issue
Block a user