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!