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

ultralytics 8.0.189 pin tensorflow<=2.13.0 to avoid tflite_support bug (#5140)

Browse source 
Co-authored-by: Shreyesh <76511222+shreyesh0610@users.noreply.github.com>
Co-authored-by: Soumik Rakshit <19soumik.rakshit96@gmail.com>
Co-authored-by: Ryan Romanchuk <rromanchuk@gmail.com>
This commit is contained in:
Glenn Jocher 2023-09-29 01:01:22 +02:00 • committed by GitHub
parent 19c3314e68
commit 2624fc04fb
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
6 changed files with 22 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

1
ultralytics/utils/callbacks/wb.py
View file

@ -54,6 +54,7 @@ def on_train_end(trainer):
if trainer.best.exists():
art.add_file(trainer.best)
wb.run.log_artifact(art, aliases=['best'])
wb.run.finish() # required or run continues on dashboard
callbacks = {