Advertisement
Guest User

Untitled

a guest
Jul 19th, 2019
79
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.23 KB | None | 0 0
  1. for node in nodes:
  2. if "LeakyRelu" in node:
  3. ns[node]=gs.create_plugin_node(name=node,op="LReLU_TRT", negSlope=0.1)
  4. if "orientation/l2_normalize" in node:
  5. dynamic_graph.remove(node)
  6. dynamic_graph.collapse_namespaces(ns)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement