A question about the detail of code

[shizia]

Hello, when I run the code, I have a question about the line 156 in the model_search.py, which is x = self.layers[i](x, self.na_weights[0], edge_index) in the loop. If the self.na_weights[0] should be self.na_weights[i] ? Because it seems that each layer has its own na_weights .