lynxkite / examples /Model definition.lynxkite.json
darabos's picture
Address review.
1f5c169
raw
history blame
14.2 kB
{
"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
}
]
}