Spaces:

chlab
/

interactive_kinematic_planet_detector

Sleeping

jpterry commited on Jun 16, 2022

Commit

fec9f2d

1 Parent(s): 709c7fc

updated model_type

Files changed (1) hide show

model_utils/efficientnet_config.py CHANGED Viewed

@@ -242,7 +242,7 @@ class FusedMBConv(nn.Module):
 class EfficientNetConfig(PretrainedConfig):
-    model_type = "efficientnet"
     def __init__(
         self,

 class EfficientNetConfig(PretrainedConfig):
+    model_type = "efficientnet_61_planet_detection"
     def __init__(
         self,