text: Title: [BUG] No such file or directory
  Body: ### Description
  <!--- Describe your bug in detail -->
  tc_bert_azureml.ipynb
  Error message:
  wc: /mnt/batch/tasks/shared/LS_root/jobs/gaia-ml-wks/azureml/3576038c-f399-4522-8674-383ad7cd316b/mounts/workspaceblobstore/azureml/88f85fcf-eaf4-4d58-bff2-712b35cb50aa/train0,: No such file or directory
  
  ### How do we replicate the bug?
  <!--- Please be specific as possible (use a list if needed). -->
  Just run the notebook tc_bert_azureml
  <!--- For example: -->
  <!--- * Create a conda environment for gpu -->
  <!--- * Run unit test `test_timer.py` -->
  <!--- * ... -->
  
  
  ### Expected behavior (i.e. solution)
  <!--- For example:  -->
  <!--- * The tests for the timer should pass successfully. -->
  
  
  ### Other Comments
  