Disty0's picture
Use nn.Linear instead of a custom linear function
f740bae verified