* update * update * update readme * update * update * update model desc * update * Update training/prepare_dataset.py