Browse Source

refactor(functional): import all from metric in nn

GitOrigin-RevId: 41ab78d78d
release-1.6
Megvii Engine Team 3 years ago
parent
commit
8796586b89
2 changed files with 5 additions and 0 deletions
  1. +4
    -0
      imperative/python/megengine/functional/metric.py
  2. +1
    -0
      imperative/python/megengine/functional/nn.py

+ 4
- 0
imperative/python/megengine/functional/metric.py View File

@@ -15,6 +15,10 @@ from .elemwise import abs, maximum, minimum
from .math import topk as _topk
from .tensor import broadcast_to, transpose

__all__ = [
"topk_accuracy",
]


def topk_accuracy(
logits: Tensor, target: Tensor, topk: Union[int, Iterable[int]] = 1


+ 1
- 0
imperative/python/megengine/functional/nn.py View File

@@ -1660,3 +1660,4 @@ warp_perspective = deprecated_func(
)
from .quantized import conv_bias_activation # isort:skip
from .loss import * # isort:skip
from .metric import * # isort:skip

Loading…
Cancel
Save