fbcode/pytorch/benchmark/torchbenchmark/models/nvidia_deeprecommender/reco_encoder/model/model.py

generatedunixname537391475639613 · meta-codesync[bot] · commit 9a5a71c6c03b · 2025-11-11T10:26:51.000-08:00
Reviewed By: xuzhao9

Differential Revision: D86652535

fbshipit-source-id: 1932902bd81d5e174d2b71b452abefb915a1431f
diff --git a/torchbenchmark/models/nvidia_deeprecommender/reco_encoder/model/model.py b/torchbenchmark/models/nvidia_deeprecommender/reco_encoder/model/model.py
@@ -72,7 +72,7 @@ def __init__(
                 for i in range(len(layer_sizes) - 1)
             ]
         )
-        for ind, w in enumerate(self.encode_w):
+        for w in self.encode_w:
             weight_init.xavier_uniform_(w)
 
         self.encode_b = nn.ParameterList(
@@ -94,7 +94,7 @@ def __init__(
                     for i in range(len(reversed_enc_layers) - 1)
                 ]
             )
-            for ind, w in enumerate(self.decode_w):
+            for w in self.decode_w:
                 nn.init.xavier_uniform_(w)
 
         self.decode_b = nn.ParameterList(

Original file line number	Diff line number	Diff line change
`@@ -72,7 +72,7 @@ def __init__(`
`72`	`72`	`for i in range(len(layer_sizes) - 1)`
`73`	`73`	`]`
`74`	`74`	`)`
`75`		`- for ind, w in enumerate(self.encode_w):`
	`75`	`+ for w in self.encode_w:`
`76`	`76`	`weight_init.xavier_uniform_(w)`
`77`	`77`
`78`	`78`	`self.encode_b = nn.ParameterList(`
`@@ -94,7 +94,7 @@ def __init__(`
`94`	`94`	`for i in range(len(reversed_enc_layers) - 1)`
`95`	`95`	`]`
`96`	`96`	`)`
`97`		`- for ind, w in enumerate(self.decode_w):`
	`97`	`+ for w in self.decode_w:`
`98`	`98`	`nn.init.xavier_uniform_(w)`
`99`	`99`
`100`	`100`	`self.decode_b = nn.ParameterList(`