train_kaggle.ipynb

{"cells":[{"attachments":{},"cell_type":"markdown","metadata":{},"source":["<a href=\"https://www.kaggle.com/code/davidfmora/deepglobe-train/notebook\" target=\"_blank\"><img align=\"left\" alt=\"Kaggle\" title=\"Open in Kaggle\" src=\"https://kaggle.com/static/images/open-in-kaggle.svg\"></a>"]},{"attachments":{},"cell_type":"markdown","metadata":{},"source":["# **Deepglobe Training**"]},{"attachments":{},"cell_type":"markdown","metadata":{},"source":["## Setup"]},{"attachments":{},"cell_type":"markdown","metadata":{},"source":["Download repository, install dependencies and move the data:"]},{"cell_type":"code","execution_count":null,"metadata":{"trusted":true},"outputs":[],"source":["old_data_dir = \"/kaggle/input/deepglobe-land-cover-classification-procesed/data/\"\n","new_data_dir = \"/kaggle/working/landcover-segmentation/data\"\n","\n","! pip install segmentation_models_pytorch -q\n","! git clone https://github.com/DavidFM43/landcover-segmentation.git\n","! mv \"$old_data_dir\"* \"$new_data_dir\" 2>/dev/null"]},{"attachments":{},"cell_type":"markdown","metadata":{},"source":["Move to the root of the repository:"]},{"cell_type":"code","execution_count":null,"metadata":{"trusted":true},"outputs":[],"source":["cd /kaggle/working/landcover-segmentation"]},{"attachments":{},"cell_type":"markdown","metadata":{},"source":["## **Run training script**\n","\n","**Note**: If you want to log the training to Weights and Biases you should first add their API key as a Kaggle secret. To do this, go to the \"Add-ons\" tab and click on \"Secrets\", then add a new secret with the label \"wandb_key\" and the value of your private API key. After that is done simple add the `--log` when running the `train.py` script."]},{"cell_type":"code","execution_count":null,"metadata":{"trusted":true},"outputs":[],"source":["! python3 train.py --config config.yaml "]}],"metadata":{"kernelspec":{"display_name":"Python 3","language":"python","name":"python3"},"language_info":{"codemirror_mode":{"name":"ipython","version":3},"file_extension":".py","mimetype":"text/x-python","name":"python","nbconvert_exporter":"python","pygments_lexer":"ipython3","version":"3.9.16"}},"nbformat":4,"nbformat_minor":4}