Updated docs of the prepare_data.py interface.
Added --{input,output}-dir args to train.py.