Spaces:
Runtime error
Runtime error
Arnaudding001
commited on
Commit
•
648ad35
1
Parent(s):
fc83017
Update encoder_encoders_model_irse.py
Browse files
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
|
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)
|