add HRFormer to BASE_TYPES

vita-epfl · Sep 29, 2023 · e6a695e · e6a695e
1 parent c827123
commit e6a695e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/openpifpaf/network/factory.py b/src/openpifpaf/network/factory.py
@@ -77,6 +77,7 @@
 CHECKPOINT_URLS = {}
 
 BASE_TYPES = set([
+    basenetworks.HRFormer,
     basenetworks.MobileNetV2,
     basenetworks.MobileNetV3,
     basenetworks.Resnet,