If you have the inputs with their corresponding targets, I would just start with a simple model using two linear layers and a nonlinearity between them, e.g. F.relu.
I’m currently not sure, what kind of number should be detected, so could you explain your use case a bit more?