Skip to content

Commit

Permalink
Include target in output_directory when previous step uses unlabeled …
Browse files Browse the repository at this point in the history
…target

Even if current pre-training phase doesn't use the target dataset, the output directory should be differentiated by target dataset
if any previous pre-training stage involved the target dataset (put or pmsl)
  • Loading branch information
itsnamgyu committed May 9, 2022
1 parent 56f8ba1 commit dddb997
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion paths.py
Original file line number Diff line number Diff line change
Expand Up @@ -88,7 +88,7 @@ def get_output_directory(params: Namespace, previous=False, makedirs=True):
if params.put or params.pmsl:
path = os.path.join(path, DATASET_KEYS[params.target_dataset])
else:
if params.ut:
if params.put or params.pmsl or params.ut:
path = os.path.join(path, DATASET_KEYS[params.target_dataset])

if makedirs:
Expand Down

0 comments on commit dddb997

Please sign in to comment.