diff --git a/torch_xla/experimental/fori_loop.py b/torch_xla/experimental/fori_loop.py index 8b5dc7efcc0..3d0fff3d49f 100644 --- a/torch_xla/experimental/fori_loop.py +++ b/torch_xla/experimental/fori_loop.py @@ -116,7 +116,8 @@ def new_body_fn(*carried_inputs): # res.extend(additional_inputs) # print("res: ", res) # return list(body_fn(*carried_inputs)).extend(additional_inputs) - self.named_parameters + # self.named_parameters + weight = self.linear.weight res = list(body_fn(*carried_inputs)) # print("res: ", res) # trynewres = res[:-1] + [res[-1]]