4 ms·

How you are going to run the model after training ? That should help you choose the instance type