π§ [Add] TODO for training without Aux branch
Browse files
yolo/tools/loss_functions.py
CHANGED
@@ -134,6 +134,7 @@ class DualLoss:
|
|
134 |
|
135 |
|
136 |
def create_loss_function(cfg: Config, vec2box) -> DualLoss:
|
|
|
137 |
loss_function = DualLoss(cfg, vec2box)
|
138 |
logger.info("β
Success load loss function")
|
139 |
return loss_function
|
|
|
134 |
|
135 |
|
136 |
def create_loss_function(cfg: Config, vec2box) -> DualLoss:
|
137 |
+
# TODO: make it flexible, if cfg doesn't contain aux, only use SingleLoss
|
138 |
loss_function = DualLoss(cfg, vec2box)
|
139 |
logger.info("β
Success load loss function")
|
140 |
return loss_function
|