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:
parent
19c3314e68
commit
2624fc04fb
6 changed files with 22 additions and 20 deletions
|
|
@ -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 = {
|
||||
|
|
|
|||
|
|
@ -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'
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue