Spaces:
Sleeping
Sleeping
Update swin_transformer_3d.py
Browse files- swin_transformer_3d.py +1 -1
swin_transformer_3d.py
CHANGED
@@ -9,7 +9,7 @@ from functools import reduce, lru_cache
|
|
9 |
from operator import mul
|
10 |
from einops import rearrange
|
11 |
|
12 |
-
from
|
13 |
|
14 |
|
15 |
class residual_feature_generator(nn.Module):
|
|
|
9 |
from operator import mul
|
10 |
from einops import rearrange
|
11 |
|
12 |
+
from submodules import ResidualBlock
|
13 |
|
14 |
|
15 |
class residual_feature_generator(nn.Module):
|