fuck this shit
Browse files
app.py
CHANGED
@@ -251,7 +251,7 @@ def predict_and_analyze(model_name, num_channels, dim, input_channel, image):
|
|
251 |
depth_mult=hparams.depth_mult,
|
252 |
)
|
253 |
# EfficientNetConfig.model_type = "efficientnet_%s_planet_detection" % (hparams.num_channels)
|
254 |
-
EfficientNetConfig.model_type = hparams.model_type
|
255 |
|
256 |
config.save_pretrained(save_directory=model_loading_name)
|
257 |
|
|
|
251 |
depth_mult=hparams.depth_mult,
|
252 |
)
|
253 |
# EfficientNetConfig.model_type = "efficientnet_%s_planet_detection" % (hparams.num_channels)
|
254 |
+
# EfficientNetConfig.model_type = hparams.model_type
|
255 |
|
256 |
config.save_pretrained(save_directory=model_loading_name)
|
257 |
|