1
|
{"format": "torch", "nodes": [{"name": "layers.0", "id": 140037239628944, "class_name": "Linear(in_features=784, out_features=128, bias=True)", "parameters": [["weight", [128, 784]], ["bias", [128]]], "output_shape": [[2, 128]], "num_parameters": [100352, 128]}, {"name": "layers.1", "id": 140037239628752, "class_name": "ReLU(inplace=True)", "parameters": [], "output_shape": [[2, 128]], "num_parameters": []}, {"name": "layers.2", "id": 140037239521488, "class_name": "Linear(in_features=128, out_features=128, bias=True)", "parameters": [["weight", [128, 128]], ["bias", [128]]], "output_shape": [[2, 128]], "num_parameters": [16384, 128]}, {"name": "layers.4", "id": 140037239521584, "class_name": "Dropout(p=0.2, inplace=False)", "parameters": [], "output_shape": [[2, 128]], "num_parameters": []}, {"name": "layers.5", "id": 140037239521344, "class_name": "Linear(in_features=128, out_features=128, bias=True)", "parameters": [["weight", [128, 128]], ["bias", [128]]], "output_shape": [[2, 128]], "num_parameters": [16384, 128]}, {"name": "layers.7", "id": 140037239521392, "class_name": "Dropout(p=0.2, inplace=False)", "parameters": [], "output_shape": [[2, 128]], "num_parameters": []}, {"name": "layers.8", "id": 140037239521536, "class_name": "Linear(in_features=128, out_features=62, bias=True)", "parameters": [["weight", [62, 128]], ["bias", [62]]], "output_shape": [[2, 62]], "num_parameters": [7936, 62]}], "edges": []}
|
