So I am trying to learn a control action of a robot. The input data is a 5-D vector that consists of 2-D position, 2-D velocity, and time (0-80 sec  - normalized in (0,1)) and the output is a 2-D control action.  Should I use maybe NLLLoss?
Edit: I do not have a classification problem, so maybe MSE with some other network output?