Skip to content
This repository has been archived by the owner on Jan 15, 2024. It is now read-only.

Commit

Permalink
update test
Browse files Browse the repository at this point in the history
  • Loading branch information
Chenguang Wang committed Apr 22, 2018
1 parent 12c7a33 commit df8bb2b
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions tests/unittest/test_models.py
Expand Up @@ -45,6 +45,7 @@ def test_text_models():
pretrained_dataset = pretrained_to_test.get(model_name)
model, _ = get_text_model(model_name, vocab=vocab, dataset_name=pretrained_dataset,
pretrained=pretrained_dataset is not None, root='tests/data/model/')

print(model)
if not pretrained_dataset:
model.collect_params().initialize()
Expand Down

0 comments on commit df8bb2b

Please sign in to comment.