Spaces:
Running
Running
| { | |
| "edges": [ | |
| { | |
| "id": "MSE loss 2 Optimizer 2", | |
| "source": "MSE loss 2", | |
| "sourceHandle": "output", | |
| "target": "Optimizer 2", | |
| "targetHandle": "loss" | |
| }, | |
| { | |
| "id": "Activation 1 MSE loss 2", | |
| "source": "Activation 1", | |
| "sourceHandle": "output", | |
| "target": "MSE loss 2", | |
| "targetHandle": "x" | |
| }, | |
| { | |
| "id": "Input: tensor 3 MSE loss 2", | |
| "source": "Input: tensor 3", | |
| "sourceHandle": "x", | |
| "target": "MSE loss 2", | |
| "targetHandle": "y" | |
| }, | |
| { | |
| "id": "Activation 1 Repeat 1", | |
| "source": "Activation 1", | |
| "sourceHandle": "output", | |
| "target": "Repeat 1", | |
| "targetHandle": "input" | |
| }, | |
| { | |
| "id": "Input: tensor 1 Linear 1", | |
| "source": "Input: tensor 1", | |
| "sourceHandle": "x", | |
| "target": "Linear 1", | |
| "targetHandle": "x" | |
| }, | |
| { | |
| "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" | |
| } | |
| ], | |
| "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": 292.3983313429414, | |
| "y": -853.8015246037802 | |
| }, | |
| "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" | |
| ] | |
| } | |
| } | |
| }, | |
| "position": { | |
| "x": 344.0, | |
| "y": 384.0 | |
| }, | |
| "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": { | |
| "x": { | |
| "name": "x", | |
| "position": "top", | |
| "type": { | |
| "type": "tensor" | |
| } | |
| } | |
| }, | |
| "params": { | |
| "name": { | |
| "default": null, | |
| "name": "name", | |
| "type": { | |
| "type": "None" | |
| } | |
| } | |
| }, | |
| "position": { | |
| "x": 258.0, | |
| "y": 397.0 | |
| }, | |
| "type": "basic" | |
| }, | |
| "params": { | |
| "name": "X" | |
| }, | |
| "status": "planned", | |
| "title": "Input: tensor" | |
| }, | |
| "dragHandle": ".bg-primary", | |
| "height": 200.0, | |
| "id": "Input: tensor 1", | |
| "position": { | |
| "x": 85.83561484252238, | |
| "y": 293.6278596776366 | |
| }, | |
| "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": { | |
| "x": { | |
| "name": "x", | |
| "position": "top", | |
| "type": { | |
| "type": "tensor" | |
| } | |
| } | |
| }, | |
| "params": { | |
| "name": { | |
| "default": null, | |
| "name": "name", | |
| "type": { | |
| "type": "None" | |
| } | |
| } | |
| }, | |
| "position": { | |
| "x": 1169.0, | |
| "y": 340.0 | |
| }, | |
| "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": { | |
| "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": {}, | |
| "position": { | |
| "x": 937.0, | |
| "y": 270.0 | |
| }, | |
| "type": "basic" | |
| }, | |
| "params": {}, | |
| "status": "planned", | |
| "title": "MSE loss" | |
| }, | |
| "dragHandle": ".bg-primary", | |
| "height": 200.0, | |
| "id": "MSE loss 2", | |
| "position": { | |
| "x": 309.4422414664647, | |
| "y": -552.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'>" | |
| } | |
| } | |
| }, | |
| "position": { | |
| "x": 487.0, | |
| "y": 443.0 | |
| }, | |
| "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": 359.0, | |
| "y": 310.0 | |
| }, | |
| "type": "basic" | |
| }, | |
| "params": { | |
| "output_dim": "4" | |
| }, | |
| "status": "planned", | |
| "title": "Linear" | |
| }, | |
| "dragHandle": ".bg-primary", | |
| "height": 200.0, | |
| "id": "Linear 1", | |
| "position": { | |
| "x": 88.83370222907377, | |
| "y": 48.642890099180136 | |
| }, | |
| "type": "basic", | |
| "width": 200.0 | |
| } | |
| ] | |
| } | |