Skip to content

GitLab

Commit f382a8de authored by Luran He's avatar Luran He Committed by Lysandre Debut
Browse files

convert int to str before adding to a str

parent 6596e3d5
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/run_lm_finetuning.py
View file @ f382a8de
...@@ -66,7 +66,7 @@ class TextDataset(Dataset): ...@@ -66,7 +66,7 @@ class TextDataset(Dataset):
def __init__(self, tokenizer, file_path='train', block_size=512): def __init__(self, tokenizer, file_path='train', block_size=512):
assert os.path.isfile(file_path) assert os.path.isfile(file_path)
directory, filename = os.path.split(file_path) directory, filename = os.path.split(file_path)
cached_features_file = os.path.join(directory, 'cached_lm_' + block_size + '_' + filename) cached_features_file = os.path.join(directory, 'cached_lm_' + str(block_size) + '_' + filename)
if os.path.exists(cached_features_file): if os.path.exists(cached_features_file):
logger.info("Loading features from cached file %s", cached_features_file) logger.info("Loading features from cached file %s", cached_features_file)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment