diff --git a/backbone/efficientnet.py b/backbone/efficientnet.py index bf79dcd..4096539 100644 --- a/backbone/efficientnet.py +++ b/backbone/efficientnet.py @@ -387,7 +387,7 @@ def efficientnet(input_size, embedding_size=512, version='b1', **kwargs): elif version == "b4": in_size = (190, 190) depth_factor = 1.8 - width_factor = 1.4 + width_factor = 1.5 dropout_rate = 0.4 elif version == "b5": in_size = (228, 228) @@ -455,4 +455,4 @@ def efficientnet(input_size, embedding_size=512, version='b1', **kwargs): embedding_size=embedding_size, **kwargs) - return net \ No newline at end of file + return net