ignore upsample nearest in flop counting
Summary: discussed in earlier diff
@@ -24,28 +24,29 @@ ACTIVATIONS_MODE = "activations"
# some extra ops to ignore from counting.
_IGNORED_OPS = [
+ "aten::add",
+ "aten::add_",
"aten::batch_norm",
+ "aten::constant_pad_nd",
"aten::div",
"aten::div_",
+ "aten::exp",
+ "aten::log2",
+ "aten::max_pool2d",
"aten::meshgrid",
- "aten::rsub",
- "aten::sub",
- "aten::relu_",
- "aten::add_",
"aten::mul",
- "aten::add",
+ "aten::mul_",
+ "aten::nonzero_numpy",
"aten::relu",
+ "aten::relu_",
+ "aten::rsub",
"aten::sigmoid",
"aten::sigmoid_",
+ "aten::softmax",
"aten::sort",
- "aten::exp",
- "aten::mul_",
- "aten::max_pool2d",
- "aten::constant_pad_nd",
"aten::sqrt",
- "aten::softmax",
- "aten::log2",
- "aten::nonzero_numpy",
+ "aten::sub",
+ "aten::upsample_nearest2d",
"prim::PythonOp",
"torchvision::nms",
]
