embeddings resizing + tie_weights

This commit is contained in:
thomwolf committed 2019-07-12 00:02:49 +02:00
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):