Arnaudding001 commited on
Commit
648ad35
1 Parent(s): fc83017

Update encoder_encoders_model_irse.py

Browse files
Files changed (1) hide show
  1. encoder_encoders_model_irse.py +1 -1
encoder_encoders_model_irse.py CHANGED
@@ -1,5 +1,5 @@
1
  from torch.nn import Linear, Conv2d, BatchNorm1d, BatchNorm2d, PReLU, Dropout, Sequential, Module
2
- from model.encoder.encoders.helpers import get_blocks, Flatten, bottleneck_IR, bottleneck_IR_SE, l2_norm
3
 
4
  """
5
  Modified Backbone implementation from [TreB1eN](https://github.com/TreB1eN/InsightFace_Pytorch)
 
1
  from torch.nn import Linear, Conv2d, BatchNorm1d, BatchNorm2d, PReLU, Dropout, Sequential, Module
2
+ from encoder_encoders_helpers import get_blocks, Flatten, bottleneck_IR, bottleneck_IR_SE, l2_norm
3
 
4
  """
5
  Modified Backbone implementation from [TreB1eN](https://github.com/TreB1eN/InsightFace_Pytorch)