diff --git a/kaldi_data_generator/main.py b/kaldi_data_generator/main.py index 2a73d18d85b3f83f58063569a0cb368d56e1d8a2..e1dadc955525553c87ac7f30701aaaa1571bcb97 100644 --- a/kaldi_data_generator/main.py +++ b/kaldi_data_generator/main.py @@ -2,6 +2,7 @@ # -*- coding: utf-8 -*- import argparse +import getpass import os import random from collections import Counter @@ -620,6 +621,8 @@ class KaldiPartitionSplitter: def create_parser(): + user_name = getpass.getuser() + parser = argparse.ArgumentParser( description="Script to generate Kaldi or kraken training data from annotations from Arkindex", formatter_class=argparse.ArgumentDefaultsHelpFormatter, @@ -802,7 +805,7 @@ def create_parser(): parser.add_argument( "--cache_dir", type=Path, - default=Path("/tmp/kaldi_data_generator/cache/"), + default=Path(f"/tmp/kaldi_data_generator_{user_name}/cache/"), help="Cache dir where to save the full size downloaded images. Change it to force redownload.", )