13 lines
381 B
Python
Executable file
13 lines
381 B
Python
Executable file
# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved.
|
|
# Copyright (c) 2018, NVIDIA CORPORATION. All rights reserved.
|
|
# from ._utils import _C
|
|
from maskrcnn_benchmark import _C
|
|
|
|
from torch.cuda.amp import custom_fwd
|
|
|
|
# Only valid with fp32 inputs - give AMP the hint
|
|
nms = custom_fwd(_C.nms)
|
|
|
|
# nms.__doc__ = """
|
|
# This function performs Non-maximum suppresion"""
|