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

ultralytics 8.0.149 add Open Images V7 training (#4178)

Browse source 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: AdiEcho <30563671+AdiEcho@users.noreply.github.com>
This commit is contained in:
Glenn Jocher 2023-08-07 01:19:51 +02:00 • committed by GitHub
parent c751c7f88a
commit 7565210484
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
110 changed files with 969 additions and 144 deletions
Download patch file Download diff file Expand all files Collapse all files

9
ultralytics/nn/tasks.py
View file

@ -161,6 +161,7 @@ class BaseModel(nn.Module):
Prints model information
Args:
detailed (bool): if True, prints out detailed information about the model. Defaults to False
verbose (bool): if True, prints out the model information. Defaults to False
imgsz (int): the size of the image that the model will be trained on. Defaults to 640
"""
@ -168,11 +169,10 @@ class BaseModel(nn.Module):
def _apply(self, fn):
"""
`_apply()` is a function that applies a function to all the tensors in the model that are not
parameters or registered buffers
Applies a function to all the tensors in the model that are not parameters or registered buffers.
Args:
fn: the function to apply to the model
fn (function): the function to apply to the model
Returns:
A model that is a Detect() object.
@ -186,7 +186,8 @@ class BaseModel(nn.Module):
return self
def load(self, weights, verbose=True):
"""Load the weights into the model.
"""
Load the weights into the model.
Args:
weights (dict | torch.nn.Module): The pre-trained weights to be loaded.