Spaces:
Sleeping
Sleeping
🧐 [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
|