From 9d7d9b3ae053da8821521c4925932bc6f79c2814 Mon Sep 17 00:00:00 2001 From: Victor SANH Date: Fri, 29 May 2020 01:10:41 -0400 Subject: [PATCH] weird import --- examples/movement-pruning/emmental/modeling_bert_masked.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/examples/movement-pruning/emmental/modeling_bert_masked.py b/examples/movement-pruning/emmental/modeling_bert_masked.py index 83a8f7c630..f2a045fe53 100644 --- 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,