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

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

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 = {

View file

@ -135,7 +135,7 @@ def select_device(device='', batch=0, newline=False, verbose=True):
p = torch.cuda.get_device_properties(i)
s += f"{'' if i == 0 else space}CUDA:{d} ({p.name}, {p.total_memory / (1 << 20):.0f}MiB)\n" # bytes to MB
arg = 'cuda:0'
elif mps and getattr(torch, 'has_mps', False) and torch.backends.mps.is_available() and TORCH_2_0:
elif mps and TORCH_2_0 and torch.backends.mps.is_available():
# Prefer MPS if available
s += f'MPS ({get_cpu_info()})\n'
arg = 'mps'