From c42091a830b6b18ec45a9bf2fff78ee28a22b0f3 Mon Sep 17 00:00:00 2001
From: Lutz Roeder <lutzroeder@users.noreply.github.com>
Date: Tue, 27 Aug 2024 17:14:56 +0200
Subject: [PATCH] Update pytorch.js

---
 source/pytorch.js | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/source/pytorch.js b/source/pytorch.js
index 21ed41bfe4..205e295640 100644
--- a/source/pytorch.js
+++ b/source/pytorch.js
@@ -3490,14 +3490,15 @@ pytorch.Utility = class {
             const modules = new Map();
             const entries = obj instanceof Map ? Array.from(obj) : Object.entries(obj);
             if (entries.length > 0 && entries) {
-                for (const [name, value] of entries) {
+                for (const [key, value] of entries) {
+                    const name = key.toString();
                     if (!value || Object(value) !== value || pytorch.Utility.isTensor(value) || ArrayBuffer.isView(value)) {
                         return null;
                     }
                     if (!modules.has(name)) {
                         modules.set(name, {});
                     }
-                    const layer = modules.get(name);
+                    const module = modules.get(name);
                     let tensor = false;
                     const entries = value instanceof Map ? value : new Map(Object.entries(value));
                     for (const [name, value] of entries) {
@@ -3511,12 +3512,12 @@ pytorch.Utility = class {
                             continue;
                         }
                         if (typeof value === 'string' || typeof value === 'number') {
-                            layer[name] = value;
+                            module[name] = value;
                             continue;
                         }
                         if (pytorch.Utility.isTensor(value)) {
                             value.__name__ = name;
-                            layer[name] = value;
+                            module[name] = value;
                             tensor = true;
                         }
                     }