Pārlūkot izejas kodu

added chainercv2.resnet18 model to supported models

Dimitri Korsch 3 gadi atpakaļ
vecāks
revīzija
c24435da08
1 mainītis faili ar 1 papildinājumiem un 1 dzēšanām
  1. 1 1
      cvmodelz/models/factory.py

+ 1 - 1
cvmodelz/models/factory.py

@@ -27,7 +27,7 @@ class ModelFactory(abc.ABC):
 
 		chainercv2=(
 			cv2resnet.resnet50,
-			# cv2resnet.resnet50b,
+			cv2resnet.resnet18,
 
 			cv2inceptionv3.inceptionv3,
 		),