Explorar o código

Update coreml.js

Lutz Roeder %!s(int64=4) %!d(string=hai) anos
pai
achega
b51e6eb26f
Modificáronse 1 ficheiros con 2 adicións e 2 borrados
  1. 2 2
      source/coreml.js

+ 2 - 2
source/coreml.js

@@ -359,7 +359,7 @@ coreml.Graph = class {
             case 'neuralNetworkClassifier': {
                 const neuralNetworkClassifier = model.neuralNetworkClassifier;
                 for (const layer of neuralNetworkClassifier.layers) {
-                    this._createNode(scope, group, layer.layer, layer.name, description, layer[layer.layer], layer.input, layer.output);
+                    this._createNode(scope, group, layer.layer, layer.name, group === '' ? '' : description, layer[layer.layer], layer.input, layer.output);
                 }
                 this._updateClassifierOutput(group, neuralNetworkClassifier);
                 this._updatePreprocessing(scope, group, neuralNetworkClassifier.preprocessing);
@@ -368,7 +368,7 @@ coreml.Graph = class {
             case 'neuralNetwork': {
                 const neuralNetwork = model.neuralNetwork;
                 for (const layer of neuralNetwork.layers) {
-                    this._createNode(scope, group, layer.layer, layer.name, description, layer[layer.layer], layer.input, layer.output);
+                    this._createNode(scope, group, layer.layer, layer.name, group === '' ? '' : description, layer[layer.layer], layer.input, layer.output);
                 }
                 this._updatePreprocessing(scope, group, neuralNetwork.preprocessing);
                 return 'Neural Network';