Browse Source

fix bug in test code

tags/v0.4.10
xuyige 5 years ago
parent
commit
327833a77f
1 changed files with 4 additions and 4 deletions
  1. +4
    -4
      test/models/test_bert.py

+ 4
- 4
test/models/test_bert.py View File

@@ -8,7 +8,7 @@ from fastNLP.models.bert import *
class TestBert(unittest.TestCase): class TestBert(unittest.TestCase):
def test_bert_1(self): def test_bert_1(self):
from fastNLP.core.const import Const from fastNLP.core.const import Const
from fastNLP.modules.encoder._bert import BertConfig
from fastNLP.modules.encoder.bert import BertConfig


model = BertForSequenceClassification(2, BertConfig(32000)) model = BertForSequenceClassification(2, BertConfig(32000))


@@ -23,7 +23,7 @@ class TestBert(unittest.TestCase):


def test_bert_2(self): def test_bert_2(self):
from fastNLP.core.const import Const from fastNLP.core.const import Const
from fastNLP.modules.encoder._bert import BertConfig
from fastNLP.modules.encoder.bert import BertConfig


model = BertForMultipleChoice(2, BertConfig(32000)) model = BertForMultipleChoice(2, BertConfig(32000))


@@ -38,7 +38,7 @@ class TestBert(unittest.TestCase):


def test_bert_3(self): def test_bert_3(self):
from fastNLP.core.const import Const from fastNLP.core.const import Const
from fastNLP.modules.encoder._bert import BertConfig
from fastNLP.modules.encoder.bert import BertConfig


model = BertForTokenClassification(7, BertConfig(32000)) model = BertForTokenClassification(7, BertConfig(32000))


@@ -53,7 +53,7 @@ class TestBert(unittest.TestCase):


def test_bert_4(self): def test_bert_4(self):
from fastNLP.core.const import Const from fastNLP.core.const import Const
from fastNLP.modules.encoder._bert import BertConfig
from fastNLP.modules.encoder.bert import BertConfig


model = BertForQuestionAnswering(BertConfig(32000)) model = BertForQuestionAnswering(BertConfig(32000))




Loading…
Cancel
Save