HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

YOLOv5 updates (#90)

Browse source 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
Glenn Jocher 2022-12-25 14:33:18 +01:00 • committed by GitHub
parent ebd3cfb2fd
commit 98815d560f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
27 changed files with 281 additions and 161 deletions
Download patch file Download diff file Expand all files Collapse all files

4
ultralytics/yolo/utils/callbacks/clearml.py
View file

@ -23,9 +23,9 @@ def on_train_start(trainer):
def on_val_end(trainer):
if trainer.epoch == 0:
model_info = {
"Inference speed (ms/img)": round(trainer.validator.speed[1], 1),
"Parameters": get_num_params(trainer.model),
"GFLOPs": round(get_flops(trainer.model), 1),
"Parameters": get_num_params(trainer.model)}
"Inference speed (ms/img)": round(trainer.validator.speed[1], 1)}
Task.current_task().connect(model_info, name='Model')