Skip to content

Instantly share code, notes, and snippets.

@r7vme
Created July 8, 2019 20:36
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save r7vme/e518ac5575cbce2430b54c8f68cb2be1 to your computer and use it in GitHub Desktop.
Save r7vme/e518ac5575cbce2430b54c8f68cb2be1 to your computer and use it in GitHub Desktop.
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)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment