merging to change repository base

2022-06-07 19:33:40 -04:00
parent 01703639ac
commit f70690efa8
5 changed files with 5 additions and 4 deletions
@@ -130,7 +130,7 @@ if __name__ == "__main__":
            pre_trained=True,
            freeze_layers=True,
            freeze_batch_norm=True,
-            base_model_type=ImageClassModels.EFFICIENTNET_V2B0,
+            base_model_type=ImageClassModels.EFFICIENTNET_V2S,
            dense_layer_neurons=1024,
            dropout_rate=.5,
        )