ultralytics 8.0.111 refactored model.loss() method (#2911)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Snyk bot <snyk-bot@snyk.io>
This commit is contained in:
parent
305cde69d0
commit
fd94d312da
14 changed files with 46 additions and 35 deletions
|
|
@ -12,7 +12,7 @@ import torch.nn as nn
|
|||
import torch.nn.functional as F
|
||||
from torch.nn.init import uniform_
|
||||
|
||||
__all__ = ['multi_scale_deformable_attn_pytorch', 'inverse_sigmoid']
|
||||
__all__ = 'multi_scale_deformable_attn_pytorch', 'inverse_sigmoid'
|
||||
|
||||
|
||||
def _get_clones(module, n):
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue