Added encoding size as a CL parameter in train_model.py

1 job for master in 11 minutes and 26 seconds (queued for 1 second)