Use nn.Linear instead of a custom linear function

#27

Using a custom linear function makes this model not compatible with quantization. There is no reason to use a custom function instead of nn.Linear.

Ready to merge
This branch is ready to get merged automatically.

Sign up or log in to comment