weird import

2020-05-29 01:10:41 -04:00
parent 2a3c88a659
commit 9d7d9b3ae0
1 changed files with 2 additions and 1 deletions
--- a/examples/movement-pruning/emmental/modeling_bert_masked.py
+++ b/examples/movement-pruning/emmental/modeling_bert_masked.py
@@ -26,7 +26,8 @@ import torch
 from torch import nn
 from torch.nn import CrossEntropyLoss, MSELoss

-from emmental import MaskedBertConfig, MaskedLinear
+from emmental import MaskedBertConfig
+from emmental.modules import MaskedLinear
 from transformers.file_utils import add_start_docstrings, add_start_docstrings_to_callable
 from transformers.modeling_bert import (
    ACT2FN,