Spaces:
Running
Running
| { | |
| "edges": [ | |
| { | |
| "id": "MSE loss 2 Optimizer 2", | |
| "source": "MSE loss 2", | |
| "sourceHandle": "output", | |
| "target": "Optimizer 2", | |
| "targetHandle": "loss" | |
| }, | |
| { | |
| "id": "Activation 1 Repeat 1", | |
| "source": "Activation 1", | |
| "sourceHandle": "output", | |
| "target": "Repeat 1", | |
| "targetHandle": "input" | |
| }, | |
| { | |
| "id": "Linear 1 Activation 1", | |
| "source": "Linear 1", | |
| "sourceHandle": "output", | |
| "target": "Activation 1", | |
| "targetHandle": "x" | |
| }, | |
| { | |
| "id": "Repeat 1 Linear 1", | |
| "source": "Repeat 1", | |
| "sourceHandle": "output", | |
| "target": "Linear 1", | |
| "targetHandle": "x" | |
| }, | |
| { | |
| "id": "Input: tensor 1 Linear 1", | |
| "source": "Input: tensor 1", | |
| "sourceHandle": "output", | |
| "target": "Linear 1", | |
| "targetHandle": "x" | |
| }, | |
| { | |
| "id": "Constant vector 1 Add 1", | |
| "source": "Constant vector 1", | |
| "sourceHandle": "output", | |
| "target": "Add 1", | |
| "targetHandle": "b" | |
| }, | |
| { | |
| "id": "Input: tensor 3 Add 1", | |
| "source": "Input: tensor 3", | |
| "sourceHandle": "output", | |
| "target": "Add 1", | |
| "targetHandle": "a" | |
| }, | |
| { | |
| "id": "Add 1 MSE loss 2", | |
| "source": "Add 1", | |
| "sourceHandle": "output", | |
| "target": "MSE loss 2", | |
| "targetHandle": "y" | |
| }, | |
| { | |
| "id": "Activation 1 Output 1", | |
| "source": "Activation 1", | |
| "sourceHandle": "output", | |
| "target": "Output 1", | |
| "targetHandle": "x" | |
| }, | |
| { | |
| "id": "Output 1 MSE loss 2", | |
| "source": "Output 1", | |
| "sourceHandle": "x", | |
| "target": "MSE loss 2", | |
| "targetHandle": "x" | |
| } | |
| ], | |
| "env": "PyTorch model", | |
| "nodes": [ | |
| { | |
| "data": { | |
| "__execution_delay": 0.0, | |
| "collapsed": null, | |
| "display": null, | |
| "error": null, | |
| "input_metadata": null, | |
| "meta": { | |
| "inputs": { | |
| "loss": { | |
| "name": "loss", | |
| "position": "bottom", | |
| "type": { | |
| "type": "tensor" | |
| } | |
| } | |
| }, | |
| "name": "Optimizer", | |
| "outputs": {}, | |
| "params": { | |
| "lr": { | |
| "default": 0.001, | |
| "name": "lr", | |
| "type": { | |
| "type": "<class 'float'>" | |
| } | |
| }, | |
| "type": { | |
| "default": "AdamW", | |
| "name": "type", | |
| "type": { | |
| "enum": [ | |
| "AdamW", | |
| "Adafactor", | |
| "Adagrad", | |
| "SGD", | |
| "Lion", | |
| "Paged AdamW", | |
| "Galore AdamW" | |
| ] | |
| } | |
| } | |
| }, | |
| "type": "basic" | |
| }, | |
| "params": { | |
| "lr": "0.1", | |
| "type": "SGD" | |
| }, | |
| "status": "planned", | |
| "title": "Optimizer" | |
| }, | |
| "dragHandle": ".bg-primary", | |
| "height": 250.0, | |
| "id": "Optimizer 2", | |
| "position": { | |
| "x": 359.75221367487865, | |
| "y": -1560.7604266065723 | |
| }, | |
| "type": "basic", | |
| "width": 232.0 | |
| }, | |
| { | |
| "data": { | |
| "__execution_delay": 0.0, | |
| "collapsed": null, | |
| "display": null, | |
| "error": null, | |
| "input_metadata": null, | |
| "meta": { | |
| "inputs": { | |
| "x": { | |
| "name": "x", | |
| "position": "bottom", | |
| "type": { | |
| "type": "<class 'inspect._empty'>" | |
| } | |
| } | |
| }, | |
| "name": "Activation", | |
| "outputs": { | |
| "output": { | |
| "name": "output", | |
| "position": "top", | |
| "type": { | |
| "type": "None" | |
| } | |
| } | |
| }, | |
| "params": { | |
| "type": { | |
| "default": "ReLU", | |
| "name": "type", | |
| "type": { | |
| "enum": [ | |
| "ReLU", | |
| "Leaky_ReLU", | |
| "Tanh", | |
| "Mish" | |
| ] | |
| } | |
| } | |
| }, | |
| "type": "basic" | |
| }, | |
| "params": { | |
| "type": "Leaky_ReLU" | |
| }, | |
| "status": "planned", | |
| "title": "Activation" | |
| }, | |
| "dragHandle": ".bg-primary", | |
| "height": 200.0, | |
| "id": "Activation 1", | |
| "position": { | |
| "x": 99.77615018185415, | |
| "y": -249.43925929074078 | |
| }, | |
| "type": "basic", | |
| "width": 200.0 | |
| }, | |
| { | |
| "data": { | |
| "__execution_delay": 0.0, | |
| "collapsed": null, | |
| "display": null, | |
| "error": null, | |
| "input_metadata": null, | |
| "meta": { | |
| "inputs": {}, | |
| "name": "Input: tensor", | |
| "outputs": { | |
| "output": { | |
| "name": "output", | |
| "position": "top", | |
| "type": { | |
| "type": "tensor" | |
| } | |
| } | |
| }, | |
| "params": { | |
| "name": { | |
| "default": null, | |
| "name": "name", | |
| "type": { | |
| "type": "None" | |
| } | |
| } | |
| }, | |
| "type": "basic" | |
| }, | |
| "params": { | |
| "name": "Y" | |
| }, | |
| "status": "planned", | |
| "title": "Input: tensor" | |
| }, | |
| "dragHandle": ".bg-primary", | |
| "height": 200.0, | |
| "id": "Input: tensor 3", | |
| "position": { | |
| "x": 485.8840220312055, | |
| "y": -149.86223034126274 | |
| }, | |
| "type": "basic", | |
| "width": 200.0 | |
| }, | |
| { | |
| "data": { | |
| "__execution_delay": null, | |
| "collapsed": true, | |
| "display": null, | |
| "error": null, | |
| "input_metadata": null, | |
| "meta": { | |
| "inputs": { | |
| "x": { | |
| "name": "x", | |
| "position": "bottom", | |
| "type": { | |
| "type": "<class 'inspect._empty'>" | |
| } | |
| }, | |
| "y": { | |
| "name": "y", | |
| "position": "bottom", | |
| "type": { | |
| "type": "<class 'inspect._empty'>" | |
| } | |
| } | |
| }, | |
| "name": "MSE loss", | |
| "outputs": { | |
| "output": { | |
| "name": "output", | |
| "position": "top", | |
| "type": { | |
| "type": "None" | |
| } | |
| } | |
| }, | |
| "params": {}, | |
| "type": "basic" | |
| }, | |
| "params": {}, | |
| "status": "planned", | |
| "title": "MSE loss" | |
| }, | |
| "dragHandle": ".bg-primary", | |
| "height": 200.0, | |
| "id": "MSE loss 2", | |
| "position": { | |
| "x": 362.77557479979805, | |
| "y": -1287.1056805642488 | |
| }, | |
| "type": "basic", | |
| "width": 200.0 | |
| }, | |
| { | |
| "data": { | |
| "__execution_delay": 0.0, | |
| "collapsed": null, | |
| "display": null, | |
| "error": null, | |
| "input_metadata": null, | |
| "meta": { | |
| "inputs": { | |
| "input": { | |
| "name": "input", | |
| "position": "top", | |
| "type": { | |
| "type": "tensor" | |
| } | |
| } | |
| }, | |
| "name": "Repeat", | |
| "outputs": { | |
| "output": { | |
| "name": "output", | |
| "position": "bottom", | |
| "type": { | |
| "type": "tensor" | |
| } | |
| } | |
| }, | |
| "params": { | |
| "same_weights": { | |
| "default": false, | |
| "name": "same_weights", | |
| "type": { | |
| "type": "<class 'bool'>" | |
| } | |
| }, | |
| "times": { | |
| "default": 1.0, | |
| "name": "times", | |
| "type": { | |
| "type": "<class 'int'>" | |
| } | |
| } | |
| }, | |
| "type": "basic" | |
| }, | |
| "params": { | |
| "same_weights": false, | |
| "times": "2" | |
| }, | |
| "status": "planned", | |
| "title": "Repeat" | |
| }, | |
| "dragHandle": ".bg-primary", | |
| "height": 200.0, | |
| "id": "Repeat 1", | |
| "position": { | |
| "x": -210.0, | |
| "y": -135.0 | |
| }, | |
| "type": "basic", | |
| "width": 200.0 | |
| }, | |
| { | |
| "data": { | |
| "__execution_delay": 0.0, | |
| "collapsed": null, | |
| "display": null, | |
| "error": null, | |
| "input_metadata": null, | |
| "meta": { | |
| "inputs": { | |
| "x": { | |
| "name": "x", | |
| "position": "bottom", | |
| "type": { | |
| "type": "<class 'inspect._empty'>" | |
| } | |
| } | |
| }, | |
| "name": "Linear", | |
| "outputs": { | |
| "output": { | |
| "name": "output", | |
| "position": "top", | |
| "type": { | |
| "type": "None" | |
| } | |
| } | |
| }, | |
| "params": { | |
| "output_dim": { | |
| "default": 1024.0, | |
| "name": "output_dim", | |
| "type": { | |
| "type": "<class 'int'>" | |
| } | |
| } | |
| }, | |
| "position": { | |
| "x": 667.0, | |
| "y": 432.0 | |
| }, | |
| "type": "basic" | |
| }, | |
| "params": { | |
| "output_dim": "4" | |
| }, | |
| "status": "planned", | |
| "title": "Linear" | |
| }, | |
| "dragHandle": ".bg-primary", | |
| "height": 200.0, | |
| "id": "Linear 1", | |
| "position": { | |
| "x": 98.54861342271252, | |
| "y": 14.121603973834155 | |
| }, | |
| "type": "basic", | |
| "width": 200.0 | |
| }, | |
| { | |
| "data": { | |
| "__execution_delay": 0.0, | |
| "collapsed": null, | |
| "display": null, | |
| "error": null, | |
| "input_metadata": null, | |
| "meta": { | |
| "inputs": {}, | |
| "name": "Input: tensor", | |
| "outputs": { | |
| "output": { | |
| "name": "output", | |
| "position": "top", | |
| "type": { | |
| "type": "tensor" | |
| } | |
| } | |
| }, | |
| "params": { | |
| "name": { | |
| "default": null, | |
| "name": "name", | |
| "type": { | |
| "type": "None" | |
| } | |
| } | |
| }, | |
| "position": { | |
| "x": 675.0, | |
| "y": 499.0 | |
| }, | |
| "type": "basic" | |
| }, | |
| "params": { | |
| "name": "X" | |
| }, | |
| "status": "planned", | |
| "title": "Input: tensor" | |
| }, | |
| "dragHandle": ".bg-primary", | |
| "height": 200.0, | |
| "id": "Input: tensor 1", | |
| "position": { | |
| "x": 108.75735538875443, | |
| "y": 331.53404347930933 | |
| }, | |
| "type": "basic", | |
| "width": 200.0 | |
| }, | |
| { | |
| "data": { | |
| "__execution_delay": 0.0, | |
| "collapsed": null, | |
| "display": null, | |
| "error": null, | |
| "input_metadata": null, | |
| "meta": { | |
| "inputs": {}, | |
| "name": "Constant vector", | |
| "outputs": { | |
| "output": { | |
| "name": "output", | |
| "position": "top", | |
| "type": { | |
| "type": "None" | |
| } | |
| } | |
| }, | |
| "params": { | |
| "size": { | |
| "default": 1.0, | |
| "name": "size", | |
| "type": { | |
| "type": "<class 'int'>" | |
| } | |
| }, | |
| "value": { | |
| "default": 0.0, | |
| "name": "value", | |
| "type": { | |
| "type": "<class 'int'>" | |
| } | |
| } | |
| }, | |
| "position": { | |
| "x": 1061.0, | |
| "y": 239.0 | |
| }, | |
| "type": "basic" | |
| }, | |
| "params": { | |
| "size": "1", | |
| "value": "1" | |
| }, | |
| "status": "planned", | |
| "title": "Constant vector" | |
| }, | |
| "dragHandle": ".bg-primary", | |
| "height": 258.0, | |
| "id": "Constant vector 1", | |
| "position": { | |
| "x": 983.1241140187901, | |
| "y": -562.803650462906 | |
| }, | |
| "type": "basic", | |
| "width": 238.0 | |
| }, | |
| { | |
| "data": { | |
| "__execution_delay": null, | |
| "collapsed": true, | |
| "display": null, | |
| "error": null, | |
| "input_metadata": null, | |
| "meta": { | |
| "inputs": { | |
| "a": { | |
| "name": "a", | |
| "position": "bottom", | |
| "type": { | |
| "type": "<class 'inspect._empty'>" | |
| } | |
| }, | |
| "b": { | |
| "name": "b", | |
| "position": "bottom", | |
| "type": { | |
| "type": "<class 'inspect._empty'>" | |
| } | |
| } | |
| }, | |
| "name": "Add", | |
| "outputs": { | |
| "output": { | |
| "name": "output", | |
| "position": "top", | |
| "type": { | |
| "type": "None" | |
| } | |
| } | |
| }, | |
| "params": {}, | |
| "position": { | |
| "x": 1077.0, | |
| "y": 334.0 | |
| }, | |
| "type": "basic" | |
| }, | |
| "params": {}, | |
| "status": "planned", | |
| "title": "Add" | |
| }, | |
| "dragHandle": ".bg-primary", | |
| "height": 200.0, | |
| "id": "Add 1", | |
| "position": { | |
| "x": 818.5444381090571, | |
| "y": -955.5157374399466 | |
| }, | |
| "type": "basic", | |
| "width": 200.0 | |
| }, | |
| { | |
| "data": { | |
| "__execution_delay": null, | |
| "collapsed": true, | |
| "display": null, | |
| "error": null, | |
| "input_metadata": null, | |
| "meta": { | |
| "inputs": { | |
| "x": { | |
| "name": "x", | |
| "position": "bottom", | |
| "type": { | |
| "type": "tensor" | |
| } | |
| } | |
| }, | |
| "name": "Output", | |
| "outputs": { | |
| "x": { | |
| "name": "x", | |
| "position": "top", | |
| "type": { | |
| "type": "tensor" | |
| } | |
| } | |
| }, | |
| "params": {}, | |
| "position": { | |
| "x": 544.0, | |
| "y": 306.0 | |
| }, | |
| "type": "basic" | |
| }, | |
| "params": {}, | |
| "status": "planned", | |
| "title": "Output" | |
| }, | |
| "dragHandle": ".bg-primary", | |
| "height": 200.0, | |
| "id": "Output 1", | |
| "position": { | |
| "x": 185.15239170944702, | |
| "y": -733.1526319565451 | |
| }, | |
| "type": "basic", | |
| "width": 200.0 | |
| } | |
| ] | |
| } | |