JasperLinmans

Hi Zabboud, I'm working on my own implementation, so I can't comment on this exact codebase. But what I found: changing the initialisation (in this case from ` nn.init.kaiming_normal_(m.weight, mode='fan_in',...