henry000 commited on
Commit
c14cffe
Β·
1 Parent(s): 60bfaff

πŸ’¬ [Update] logging messange for building model

Browse files
Files changed (1) hide show
  1. 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 model-{arch_name}")
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", {})