Skip to content
Snippets Groups Projects
  1. Oct 03, 2018
  2. Aug 29, 2018
  3. Aug 27, 2018
  4. Jul 11, 2018
  5. May 16, 2018
    • Neta Zmora's avatar
      PyTorch 0.4 improvement to SummaryGraph · 32c01c28
      Neta Zmora authored
      PyTorch 0.4 now fully supports the ONNX export features that are needed
      in order to create a SummaryGraph, which is sort of a "shadow graph" for
      PyTorch models.
      
      The big advantage of SummaryGraph is that it gives us information about
      the connectivity of nodes.  With connectivity information we can compute
      per-node MAC (compute) and BW, and better yet, we can remove channels,
      filters, and layers (more on this in future commits).
      
      In this commit we (1) replace the long and overly-verbose ONNX node names,
      with PyTorch names; and (2) move MAC and BW attributes from the Jupyter
      notebook to the SummaryGraph.
      32c01c28
    • Neta Zmora's avatar
      pytorch 0.4: adjustments to API changes · 957e6777
      Neta Zmora authored
      Various small changes due to the chamnges in the semantics and syntax of the
      PyTorch 0.4 API.
      
      Note that currently distiller.model_performance_summary() returns wrong results
      on graphs containing torch.nn.DataParallel layers.
      957e6777
    • Neta Zmora's avatar
      b9bf4282
  6. May 10, 2018
  7. Apr 25, 2018
  8. Apr 24, 2018
Loading