Thank you for your answer. Regarding the last part, if I understand correctly, the code should be changed to:
self._fc[0].weight = torch.nn.Parameter(new_wi)

Will this update the parameters of the assigned layer with the values of “new_wi”?