Files changed (1) hide show
  1. block.py +0 -1
block.py CHANGED
@@ -12,7 +12,6 @@ import torch.nn as nn
12
  import torch.nn.functional as F
13
  from torch import Tensor
14
 
15
- from .stochastic_depth import StochasticDepth
16
  from .mha import MHA
17
  from .mlp import Mlp
18
 
 
12
  import torch.nn.functional as F
13
  from torch import Tensor
14
 
 
15
  from .mha import MHA
16
  from .mlp import Mlp
17