tensorflow / lucid

A collection of infrastructure and tools for research in neural network interpretability.
Apache License 2.0
4.65k stars 655 forks source link

graph_analysis: nicer json parsed graph structure #216

Closed colah closed 3 years ago

colah commented 4 years ago

Nicer layout parse structure for @shancarter:

"layout": {
    "type": "Sequence",
    "children": [
      {
        "type": "Node",
        "name": "input:0"
      },
      {
        "type": "Node",
        "name": "conv2d0:0"
      },
      {
        "type": "Node",
        "name": "maxpool0:0"
      },
      {
        "type": "Node",
        "name": "conv2d1:0"
      },
      {
        "type": "Node",
        "name": "conv2d2:0"
      },
      {
        "type": "Node",
        "name": "maxpool1:0"
      },
      {
        "type": "Branch",
        "children": [
          {
            "type": "Node",
            "name": "mixed3a_1x1:0"
          },
          {
            "type": "Sequence",
            "children": [
              {
                "type": "Node",
                "name": "mixed3a_3x3_bottleneck:0"
              },
              {
                "type": "Node",
                "name": "mixed3a_3x3:0"
              }
            ]
          },
...
colah commented 4 years ago

This change probably breaks some things.

google-cla[bot] commented 3 years ago

Thanks for your pull request. It looks like this may be your first contribution to a Google open source project (if not, look below for help). Before we can look at your pull request, you'll need to sign a Contributor License Agreement (CLA).

:memo: Please visit https://cla.developers.google.com/ to sign.

Once you've signed (or fixed any issues), please reply here with @googlebot I signed it! and we'll verify it.


What to do if you already signed the CLA

Individual signers
Corporate signers

ℹ️ Googlers: Go here for more info.

ludwigschubert commented 3 years ago

Sorry for the mess, everyone! I tried to rebase this on master, but ended up creating a huge merge commit. I've cherry picked @colah 's changes instead, and they're merged on master.