As shown in the error message
TypeError: new() received an invalid combination of arguments - got (google.protobuf.pyext._message.RepeatedScalarContainer, int), but expected one of:

google.protobuf.pyext._message.RepeatedScalarContainer should be an Int. As far as what it means, I am not sure, it depends on the api you are using. But the main problem is you are not giving input as Int to the layer nn.Linear