Your model works for me passing these three arguments.
However, it seems they are not used in your current model architecture, but that doesn’t explain the error.
Are you sure you’ve defined this model or do you have another class called Net?