π¬ [Update] logging messange for building model
Browse files- yolo/model/yolo.py +2 -1
yolo/model/yolo.py
CHANGED
@@ -27,8 +27,9 @@ class YOLO(nn.Module):
|
|
27 |
model_list = nn.ModuleList()
|
28 |
output_dim = [3]
|
29 |
layer_indices_by_tag = {}
|
|
|
30 |
for arch_name in model_arch:
|
31 |
-
logger.info(f"ποΈ Building
|
32 |
for layer_idx, layer_spec in enumerate(model_arch[arch_name], start=1):
|
33 |
layer_type, layer_info = next(iter(layer_spec.items()))
|
34 |
layer_args = layer_info.get("args", {})
|
|
|
27 |
model_list = nn.ModuleList()
|
28 |
output_dim = [3]
|
29 |
layer_indices_by_tag = {}
|
30 |
+
logger.info(f"π Building YOLO")
|
31 |
for arch_name in model_arch:
|
32 |
+
logger.info(f" ποΈ Building {arch_name}")
|
33 |
for layer_idx, layer_spec in enumerate(model_arch[arch_name], start=1):
|
34 |
layer_type, layer_info = next(iter(layer_spec.items()))
|
35 |
layer_args = layer_info.get("args", {})
|