Take the training notebook from colab and update to copy all the cudf updates in rapids_streetcar_DL_refactored_pipeline.ipynb in colab