Hack for base_path in train.py