Skip to content

Commit 3fb4257

Browse files
committed
fixed logging message
1 parent 3b02249 commit 3fb4257

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

notebooks/lang_model_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -285,7 +285,7 @@ def get_emb_batch(lang_model, np_array, bs, dest_dir):
285285
np.save(destPath/f'lang_model_last_emb_{i}.npy', y_max)
286286
np.save(destPath/f'lang_model_pool_emb_{i}.npy', np.concatenate([y_mean, y_max, y_last], axis=1))
287287

288-
logging.warning(f'Saved {2*len(data_chunked)} files to {str(destPath.absolute())}')
288+
logging.warning(f'Saved {4*len(data_chunked)} files to {str(destPath.absolute())}')
289289

290290

291291
def get_emb(vocab, lang_model, sentence_str, last_or_mean='mean'):

0 commit comments

Comments
 (0)