Skip to content

GitLab

Commit 3d8b5067 authored by vishnubanna's avatar vishnubanna
Browse files

final test

parent cae2622b
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
official/vision/beta/projects/yolo/modeling/backbones/Darknet.py
View file @ 3d8b5067
...@@ -352,13 +352,15 @@ def build_darknet( ...@@ -352,13 +352,15 @@ def build_darknet(
backbone_type = model_config.backbone.type backbone_type = model_config.backbone.type
backbone_cfg = model_config.backbone.get() backbone_cfg = model_config.backbone.get()
norm_activation_config = model_config.norm_activation norm_activation_config = model_config.norm_activation
return Darknet(model_id=backbone_cfg.model_id, model = Darknet(model_id=backbone_cfg.model_id,
input_shape=input_specs, input_shape=input_specs,
activation=norm_activation_config.activation, activation=norm_activation_config.activation,
use_sync_bn=norm_activation_config.use_sync_bn, use_sync_bn=norm_activation_config.use_sync_bn,
norm_momentum=norm_activation_config.norm_momentum, norm_momentum=norm_activation_config.norm_momentum,
norm_epsilon=norm_activation_config.norm_epsilon, norm_epsilon=norm_activation_config.norm_epsilon,
kernel_regularizer=l2_regularizer) kernel_regularizer=l2_regularizer)
model.summary()
return model
# if __name__ == "__main__": # if __name__ == "__main__":
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment