Removed base_dir in train_parts.py.