Python refactor and simplification (#12624)
* Python code cleanup * Auto-format by Ultralytics actions --------- Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
This commit is contained in:
1 parent
de64179cd4
commit
22e9afef74
11 files changed
+155
-172
No files matched your search
+1
-1
@@ -188,7 +188,7 @@ class Profile(contextlib.ContextDecorator):
|
||||
def __init__(self, t=0.0, device: torch.device = None):
|
||||
self.t = t
|
||||
self.device = device
|
||||
self.cuda = True if (device and str(device)[:4] == "cuda") else False
|
||||
self.cuda = bool(device and str(device).startswith("cuda"))
|
||||
|
||||
def __enter__(self):
|
||||
self.start = self.time()
|
||||
|
||||
Reference in new issue
Block a user