Spaces:
Sleeping
Sleeping
Update lxmert/src/lxmert_lrp.py
Browse files- lxmert/src/lxmert_lrp.py +1 -1
lxmert/src/lxmert_lrp.py
CHANGED
@@ -24,7 +24,7 @@ from typing import Optional, Tuple
|
|
24 |
import torch
|
25 |
from torch import nn
|
26 |
from torch.nn import CrossEntropyLoss, SmoothL1Loss
|
27 |
-
from
|
28 |
from transformers.file_utils import (
|
29 |
ModelOutput,
|
30 |
add_code_sample_docstrings,
|
|
|
24 |
import torch
|
25 |
from torch import nn
|
26 |
from torch.nn import CrossEntropyLoss, SmoothL1Loss
|
27 |
+
from .layers import *
|
28 |
from transformers.file_utils import (
|
29 |
ModelOutput,
|
30 |
add_code_sample_docstrings,
|