Skip to content

Commit 50a363c

Browse files
committed
remove results_folder option from preprocess script
1 parent e126424 commit 50a363c

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

scripts/preprocess_data.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,15 +16,14 @@
1616
if __name__ == '__main__':
1717
parser = argparse.ArgumentParser(description='train coarse stage')
1818
parser.add_argument('--stage', default='all', help='Name of the stage we want to process data for (semantic, coarse, fine, all). Provide "all" to process all stages in succession.')
19-
parser.add_argument('--results_folder', default='./data/preprocessed_fma')
2019
parser.add_argument('--model_config', default='./configs/model/musiclm_small.json')
2120
parser.add_argument('--training_config', default='./configs/training/train_fma_preprocess.json')
2221
parser.add_argument('--rvq_path', default='./checkpoints/clap.rvq.350.pt')
2322
parser.add_argument('--kmeans_path', default='./results/hubert_kmeans/kmeans.joblib')
2423

2524
args = parser.parse_args()
2625

27-
print(f'saving to {args.results_folder}, using model config {args.model_config}, training config {args.training_config}, rvq checkpoint {args.rvq_path}, kmeans checkpoint {args.kmeans_path}')
26+
print(f'using model config {args.model_config}, training config {args.training_config}, rvq checkpoint {args.rvq_path}, kmeans checkpoint {args.kmeans_path}')
2827
print(f'processing stage(s): {args.stage}')
2928

3029
model_config = load_model_config(args.model_config)

0 commit comments

Comments
 (0)