Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- for node in nodes:
- if "LeakyRelu" in node:
- ns[node]=gs.create_plugin_node(name=node,op="LReLU_TRT", negSlope=0.1)
- if "orientation/l2_normalize" in node:
- dynamic_graph.remove(node)
- dynamic_graph.collapse_namespaces(ns)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement