Skip to content

GitLab

Commit 9d7d9b3a authored by Victor SANH's avatar Victor SANH
Browse files

weird import

parent 2a3c88a6
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
examples/movement-pruning/emmental/modeling_bert_masked.py
View file @ 9d7d9b3a
...@@ -26,7 +26,8 @@ import torch ...@@ -26,7 +26,8 @@ import torch
from torch import nn from torch import nn
from torch.nn import CrossEntropyLoss, MSELoss 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.file_utils import add_start_docstrings, add_start_docstrings_to_callable
from transformers.modeling_bert import ( from transformers.modeling_bert import (
ACT2FN, ACT2FN,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment