Skip to content

Commit f1b42e2

Browse files
author
yarden-sony
committed
remove custom_layers import
1 parent 0ca1539 commit f1b42e2

File tree

1 file changed

+2
-3
lines changed
  • model_compression_toolkit/target_platform_capabilities/targetplatform2framework

1 file changed

+2
-3
lines changed

model_compression_toolkit/target_platform_capabilities/targetplatform2framework/attach2pytorch.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,6 @@
3232
from model_compression_toolkit.target_platform_capabilities.targetplatform2framework.attach2fw import \
3333
AttachTpcToFramework
3434
from model_compression_toolkit.target_platform_capabilities.targetplatform2framework.attribute_filter import Eq
35-
from sony_custom_layers.pytorch import MulticlassNMS, MulticlassNMSWithIndices, FasterRCNNBoxDecode
3635

3736

3837
class AttachTpcToPytorch(AttachTpcToFramework):
@@ -98,8 +97,8 @@ def __init__(self):
9897
OperatorSetNames.L2NORM: [LayerFilterParams(torch.nn.functional.normalize,
9998
Eq('p', 2) | Eq('p', None))],
10099
OperatorSetNames.SSD_POST_PROCESS: [], # no such operator in pytorch
101-
OperatorSetNames.COMBINED_NON_MAX_SUPPRESSION: [MulticlassNMS, MulticlassNMSWithIndices], # no such operator in pytorch
102-
OperatorSetNames.BOX_DECODE: [FasterRCNNBoxDecode]
100+
OperatorSetNames.COMBINED_NON_MAX_SUPPRESSION: [], # no such operator in pytorch
101+
OperatorSetNames.BOX_DECODE: [] # no such operator in pytorch
103102
}
104103

105104
pytorch_linear_attr_mapping = {KERNEL_ATTR: DefaultDict(default_value=PYTORCH_KERNEL),

0 commit comments

Comments
 (0)