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": { | |
"color": "green", | |
"inputs": [ | |
{ | |
"name": "loss", | |
"position": "bottom", | |
"type": { | |
"type": "tensor" | |
} | |
} | |
], | |
"name": "Optimizer", | |
"outputs": [], | |
"params": [ | |
{ | |
"default": "AdamW", | |
"name": "type", | |
"type": { | |
"enum": [ | |
"AdamW", | |
"Adafactor", | |
"Adagrad", | |
"SGD", | |
"Lion", | |
"Paged AdamW", | |
"Galore AdamW" | |
] | |
} | |
}, | |
{ | |
"default": 0.001, | |
"name": "lr", | |
"type": { | |
"type": "<class 'float'>" | |
} | |
} | |
], | |
"type": "basic" | |
}, | |
"params": { | |
"lr": "0.1", | |
"type": "SGD" | |
}, | |
"status": "done", | |
"title": "Optimizer" | |
}, | |
"dragHandle": ".bg-primary", | |
"height": 250.0, | |
"id": "Optimizer 2", | |
"position": { | |
"x": 359.75221367487865, | |
"y": -1150.2183224762075 | |
}, | |
"type": "basic", | |
"width": 232.0 | |
}, | |
{ | |
"data": { | |
"__execution_delay": 0.0, | |
"collapsed": null, | |
"display": null, | |
"error": null, | |
"input_metadata": null, | |
"meta": { | |
"color": "orange", | |
"inputs": [ | |
{ | |
"name": "x", | |
"position": "bottom", | |
"type": { | |
"type": "<class 'inspect._empty'>" | |
} | |
} | |
], | |
"name": "Activation", | |
"outputs": [ | |
{ | |
"name": "output", | |
"position": "top", | |
"type": { | |
"type": "None" | |
} | |
} | |
], | |
"params": [ | |
{ | |
"default": "ReLU", | |
"name": "type", | |
"type": { | |
"enum": [ | |
"ReLU", | |
"Leaky_ReLU", | |
"Tanh", | |
"Mish" | |
] | |
} | |
} | |
], | |
"type": "basic" | |
}, | |
"params": { | |
"type": "Leaky_ReLU" | |
}, | |
"status": "done", | |
"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": { | |
"color": "orange", | |
"inputs": [], | |
"name": "Input: tensor", | |
"outputs": [ | |
{ | |
"name": "output", | |
"position": "top", | |
"type": { | |
"type": "tensor" | |
} | |
} | |
], | |
"params": [ | |
{ | |
"default": null, | |
"name": "name", | |
"type": { | |
"type": "None" | |
} | |
} | |
], | |
"type": "basic" | |
}, | |
"params": { | |
"name": "Y" | |
}, | |
"status": "done", | |
"title": "Input: tensor" | |
}, | |
"dragHandle": ".bg-primary", | |
"height": 200.0, | |
"id": "Input: tensor 3", | |
"position": { | |
"x": 454.7823474758749, | |
"y": -212.0655794519241 | |
}, | |
"type": "basic", | |
"width": 200.0 | |
}, | |
{ | |
"data": { | |
"__execution_delay": null, | |
"collapsed": true, | |
"display": null, | |
"error": null, | |
"input_metadata": null, | |
"meta": { | |
"color": "orange", | |
"inputs": [ | |
{ | |
"name": "x", | |
"position": "bottom", | |
"type": { | |
"type": "<class 'inspect._empty'>" | |
} | |
}, | |
{ | |
"name": "y", | |
"position": "bottom", | |
"type": { | |
"type": "<class 'inspect._empty'>" | |
} | |
} | |
], | |
"name": "MSE loss", | |
"outputs": [ | |
{ | |
"name": "output", | |
"position": "top", | |
"type": { | |
"type": "None" | |
} | |
} | |
], | |
"params": [], | |
"type": "basic" | |
}, | |
"params": {}, | |
"status": "done", | |
"title": "MSE loss" | |
}, | |
"dragHandle": ".bg-primary", | |
"height": 200.0, | |
"id": "MSE loss 2", | |
"position": { | |
"x": 375.21624462193034, | |
"y": -721.0552036572305 | |
}, | |
"type": "basic", | |
"width": 200.0 | |
}, | |
{ | |
"data": { | |
"__execution_delay": 0.0, | |
"collapsed": null, | |
"display": null, | |
"error": null, | |
"input_metadata": null, | |
"meta": { | |
"color": "orange", | |
"inputs": [ | |
{ | |
"name": "input", | |
"position": "top", | |
"type": { | |
"type": "tensor" | |
} | |
} | |
], | |
"name": "Repeat", | |
"outputs": [ | |
{ | |
"name": "output", | |
"position": "bottom", | |
"type": { | |
"type": "tensor" | |
} | |
} | |
], | |
"params": [ | |
{ | |
"default": 1.0, | |
"name": "times", | |
"type": { | |
"type": "<class 'int'>" | |
} | |
}, | |
{ | |
"default": false, | |
"name": "same_weights", | |
"type": { | |
"type": "<class 'bool'>" | |
} | |
} | |
], | |
"type": "basic" | |
}, | |
"params": { | |
"same_weights": false, | |
"times": "2" | |
}, | |
"status": "done", | |
"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": { | |
"color": "blue", | |
"inputs": [ | |
{ | |
"name": "x", | |
"position": "bottom", | |
"type": { | |
"type": "<class 'inspect._empty'>" | |
} | |
} | |
], | |
"name": "Linear", | |
"outputs": [ | |
{ | |
"name": "output", | |
"position": "top", | |
"type": { | |
"type": "None" | |
} | |
} | |
], | |
"params": [ | |
{ | |
"default": 1024.0, | |
"name": "output_dim", | |
"type": { | |
"type": "<class 'int'>" | |
} | |
} | |
], | |
"type": "basic" | |
}, | |
"params": { | |
"output_dim": "4" | |
}, | |
"status": "done", | |
"title": "Linear" | |
}, | |
"dragHandle": ".bg-primary", | |
"height": 189.0, | |
"id": "Linear 1", | |
"position": { | |
"x": 98.54861342271252, | |
"y": 14.121603973834155 | |
}, | |
"type": "basic", | |
"width": 199.0 | |
}, | |
{ | |
"data": { | |
"__execution_delay": 0.0, | |
"collapsed": null, | |
"display": null, | |
"error": null, | |
"input_metadata": null, | |
"meta": { | |
"color": "orange", | |
"inputs": [], | |
"name": "Input: tensor", | |
"outputs": [ | |
{ | |
"name": "output", | |
"position": "top", | |
"type": { | |
"type": "tensor" | |
} | |
} | |
], | |
"params": [ | |
{ | |
"default": null, | |
"name": "name", | |
"type": { | |
"type": "None" | |
} | |
} | |
], | |
"type": "basic" | |
}, | |
"params": { | |
"name": "X" | |
}, | |
"status": "done", | |
"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": { | |
"color": "orange", | |
"inputs": [], | |
"name": "Constant vector", | |
"outputs": [ | |
{ | |
"name": "output", | |
"position": "top", | |
"type": { | |
"type": "None" | |
} | |
} | |
], | |
"params": [ | |
{ | |
"default": 0.0, | |
"name": "value", | |
"type": { | |
"type": "<class 'int'>" | |
} | |
}, | |
{ | |
"default": 1.0, | |
"name": "size", | |
"type": { | |
"type": "<class 'int'>" | |
} | |
} | |
], | |
"type": "basic" | |
}, | |
"params": { | |
"size": "1", | |
"value": "1" | |
}, | |
"status": "done", | |
"title": "Constant vector" | |
}, | |
"dragHandle": ".bg-primary", | |
"height": 258.0, | |
"id": "Constant vector 1", | |
"position": { | |
"x": 846.2767459753351, | |
"y": -226.90556526533476 | |
}, | |
"type": "basic", | |
"width": 238.0 | |
}, | |
{ | |
"data": { | |
"__execution_delay": null, | |
"collapsed": true, | |
"display": null, | |
"error": null, | |
"input_metadata": null, | |
"meta": { | |
"color": "orange", | |
"inputs": [ | |
{ | |
"name": "a", | |
"position": "bottom", | |
"type": { | |
"type": "<class 'inspect._empty'>" | |
} | |
}, | |
{ | |
"name": "b", | |
"position": "bottom", | |
"type": { | |
"type": "<class 'inspect._empty'>" | |
} | |
} | |
], | |
"name": "Add", | |
"outputs": [ | |
{ | |
"name": "output", | |
"position": "top", | |
"type": { | |
"type": "None" | |
} | |
} | |
], | |
"params": [], | |
"type": "basic" | |
}, | |
"params": {}, | |
"status": "done", | |
"title": "Add" | |
}, | |
"dragHandle": ".bg-primary", | |
"height": 200.0, | |
"id": "Add 1", | |
"position": { | |
"x": 631.934390777073, | |
"y": -395.6855954439944 | |
}, | |
"type": "basic", | |
"width": 200.0 | |
}, | |
{ | |
"data": { | |
"__execution_delay": null, | |
"collapsed": true, | |
"display": null, | |
"error": null, | |
"input_metadata": null, | |
"meta": { | |
"color": "orange", | |
"inputs": [ | |
{ | |
"name": "x", | |
"position": "bottom", | |
"type": { | |
"type": "tensor" | |
} | |
} | |
], | |
"name": "Output", | |
"outputs": [ | |
{ | |
"name": "x", | |
"position": "top", | |
"type": { | |
"type": "tensor" | |
} | |
} | |
], | |
"params": [ | |
{ | |
"default": null, | |
"name": "name", | |
"type": { | |
"type": "None" | |
} | |
} | |
], | |
"type": "basic" | |
}, | |
"params": {}, | |
"status": "done", | |
"title": "Output" | |
}, | |
"dragHandle": ".bg-primary", | |
"height": 200.0, | |
"id": "Output 1", | |
"position": { | |
"x": 119.83887514325258, | |
"y": -453.23756095856885 | |
}, | |
"type": "basic", | |
"width": 200.0 | |
} | |
] | |
} | |