Update pytorch.js

lutzroeder · Aug 30, 2024 · 4dd7bcf · 4dd7bcf
1 parent 88651ea
commit 4dd7bcf
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/source/pytorch.js b/source/pytorch.js
@@ -192,6 +192,7 @@ pytorch.Graph = class {
         } else {
             const weights = this.type === 'weights' ? module : pytorch.Utility.weights(module);
             if (weights) {
+                this.name = !this.name && typeof module.__name__ === 'string' ? module.__name__ : this.name;
                 for (const [name, module] of weights) {
                     const node = new pytorch.Node(metadata, name, 'Weights', module);
                     this.nodes.push(node);