modified tfrecords_path split by / to accomodate windows path as well…

google-research / electra

ELECTRA: Pre-training Text Encoders as Discriminators Rather Than Generators

Apache License 2.0

2.33k stars 352 forks source link

modified tfrecords_path split by / to accomodate windows path as well… #55

Open prakashr85 opened 4 years ago

prakashr85 commented 4 years ago

utils.mkdir(tfrecords_path.rsplit("/", 1)[0]) this throws error while trying to save tfrecord file on windows machine as the path separator for windows is '\'. Made a simple change to accommodate windows as well. utils.mkdir(os.path.dirname(tfrecords_path))