Update prototype.py

jonatelintelo · Mar 22, 2023 · e15d5fd · e15d5fd
1 parent a09e1a1
commit e15d5fd
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/skeleton/models/prototype.py b/skeleton/models/prototype.py
@@ -24,7 +24,9 @@
     EvaluationPair,
     evaluate_speaker_trials,
 )
+
 from skeleton.layers.resnext import ResNext
+from skeleton.layers.resnet import ResNet
 
 from skeleton.layers.statistical_pooling import MeanStatPool1D