Grégoire Clément

Results 3 comments of Grégoire Clément

This should fix it: ```python for _, module in self.fc.named_modules(): if isinstance(module, nn.Linear): module.weight.data = F.normalize(module.weight, p=2, dim=1) ```

@Fokko: Sure, I can take the time this week to review it!