mirror of
https://github.com/huggingface/transformers.git
synced 2025-07-31 02:02:21 +06:00
convert int to str before adding to a str
This commit is contained in:
parent
6596e3d566
commit
f382a8decd
@ -66,7 +66,7 @@ class TextDataset(Dataset):
|
||||
def __init__(self, tokenizer, file_path='train', block_size=512):
|
||||
assert os.path.isfile(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):
|
||||
logger.info("Loading features from cached file %s", cached_features_file)
|
||||
|
Loading…
Reference in New Issue
Block a user