{"metadata":{"kernelspec":{"language":"python","display_name":"Python 3","name":"python3"},"language_info":{"name":"python","version":"3.10.13","mimetype":"text/x-python","codemirror_mode":{"name":"ipython","version":3},"pygments_lexer":"ipython3","nbconvert_exporter":"python","file_extension":".py"},"kaggle":{"accelerator":"gpu","dataSources":[{"sourceId":47317,"databundleVersionId":5799376,"sourceType":"competition"},{"sourceId":9297804,"sourceType":"datasetVersion","datasetId":5491536}],"dockerImageVersionId":30747,"isInternetEnabled":true,"language":"python","sourceType":"notebook","isGpuEnabled":true}},"nbformat_minor":4,"nbformat":4,"cells":[{"cell_type":"code","source":"!pip install segmentation_models_pytorch","metadata":{"_uuid":"8f2839f25d086af736a60e9eeb907d3b93b6e0e5","_cell_guid":"b1076dfc-b9ad-4769-8c92-a6c4dae69d19","execution":{"iopub.status.busy":"2024-09-02T02:33:45.520224Z","iopub.execute_input":"2024-09-02T02:33:45.520672Z","iopub.status.idle":"2024-09-02T02:34:07.197615Z","shell.execute_reply.started":"2024-09-02T02:33:45.520642Z","shell.execute_reply":"2024-09-02T02:34:07.196690Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"!pip install warmup_scheduler","metadata":{"execution":{"iopub.status.busy":"2024-09-02T02:34:07.199582Z","iopub.execute_input":"2024-09-02T02:34:07.199885Z","iopub.status.idle":"2024-09-02T02:34:22.288103Z","shell.execute_reply.started":"2024-09-02T02:34:07.199859Z","shell.execute_reply":"2024-09-02T02:34:22.287169Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"import sys\nsys.path.insert(1, \"/kaggle/input/vesuvius-training-functions\") \nimport os\nimport configs\nimport training_functions","metadata":{"execution":{"iopub.status.busy":"2024-09-02T02:34:22.289632Z","iopub.execute_input":"2024-09-02T02:34:22.290028Z","iopub.status.idle":"2024-09-02T02:34:31.060845Z","shell.execute_reply.started":"2024-09-02T02:34:22.289993Z","shell.execute_reply":"2024-09-02T02:34:31.059860Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"cfg = configs.CFG()\ncfg","metadata":{"execution":{"iopub.status.busy":"2024-09-02T02:34:31.062808Z","iopub.execute_input":"2024-09-02T02:34:31.063271Z","iopub.status.idle":"2024-09-02T02:34:31.070757Z","shell.execute_reply.started":"2024-09-02T02:34:31.063246Z","shell.execute_reply":"2024-09-02T02:34:31.069554Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"#Changing configs params\ncfg.n_epochs = 3\ncfg.checkpoint = \"tu-resnet18\"\ncfg.frag_sel = [\"1\"]\ncfg.img_size = 224\ncfg.augment = \"public_baseline\"\ncfg.frag_min = 31\ncfg.frag_len = 4\n\ncfg.framework = \"smp_unet\"","metadata":{"execution":{"iopub.status.busy":"2024-09-02T02:34:31.072173Z","iopub.execute_input":"2024-09-02T02:34:31.072450Z","iopub.status.idle":"2024-09-02T02:34:31.103423Z","shell.execute_reply.started":"2024-09-02T02:34:31.072427Z","shell.execute_reply":"2024-09-02T02:34:31.102393Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"#train model + print out df detailing training stats\nlog_df_all = training_functions.run_training(cfg)\nlog_df_all","metadata":{"execution":{"iopub.status.busy":"2024-09-02T02:34:31.104633Z","iopub.execute_input":"2024-09-02T02:34:31.104981Z"},"trusted":true},"execution_count":null,"outputs":[]}]}