Code

Refactor graph visualization#165

Code

Merged

zhiqwang merged 16 commits intomasterzhiqwang/yolort:masterfrom refactor-graphzhiqwang/yolort:refactor-graphCopy head branch name to clipboardSep 16, 2021

+935-488

Commits

Commits on Sep 16, 2021

• Move graph_utils.py to relaying

  zhiqwang

  committed
  daa06c6

• Add tracing wrapper of YOLOModule

  zhiqwang

  committed
  d9dba6c

• Fixing lint

  zhiqwang

  committed
  f1ff6bf

• Make IR visualizer as a Class

  zhiqwang

  committed
  094096d

• Reset the unseen_ops

  zhiqwang

  committed
  f508574

• Refactor the dot initialization

  zhiqwang

  committed
  bbaa13c

• Adopt the f-strings

  zhiqwang

  committed
  6797228

• Make get_node_names explicit

  zhiqwang

  committed
  24a12f3

• Adding minor ultis

  zhiqwang

  committed
  fbcb672

• Move predictions to attributes

  zhiqwang

  committed
  8bd2673

• Move self.predictions above the make_graph

  zhiqwang

  committed
  31698fc

• Fix status initialization

  zhiqwang

  committed
  a084845

• Add get_trace_module unit-test

  zhiqwang

  committed
  01a852a

• Update tvm relay and graph visualize notebooks

  zhiqwang

  committed
  f9e8a54

• Add more assertions in unit-test

  zhiqwang

  committed
  796b100

• Add docstring for get_trace_module

  zhiqwang

  committed
  f9d1bc2