Skip to content

GitLab

"...text-generation-inference.git" did not exist on "ed2a3f617e6116c1fe8ea79aead4fd89b8140a6f"
Unverified Commit 1c7aa0c0 authored by PatrickBue's avatar PatrickBue Committed by GitHub
Browse files

added return statement (#2029)

parent 561a014b
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
references/detection/engine.py
View file @ 1c7aa0c0
...@@ -52,6 +52,8 @@ def train_one_epoch(model, optimizer, data_loader, device, epoch, print_freq): ...@@ -52,6 +52,8 @@ def train_one_epoch(model, optimizer, data_loader, device, epoch, print_freq):
metric_logger.update(loss=losses_reduced, **loss_dict_reduced) metric_logger.update(loss=losses_reduced, **loss_dict_reduced)
metric_logger.update(lr=optimizer.param_groups[0]["lr"]) metric_logger.update(lr=optimizer.param_groups[0]["lr"])
return metric_logger
def _get_iou_types(model): def _get_iou_types(model):
model_without_ddp = model model_without_ddp = model
......
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