{"metadata":{"kernelspec":{"language":"python","display_name":"Python 3","name":"python3"},"language_info":{"pygments_lexer":"ipython3","nbconvert_exporter":"python","version":"3.6.4","file_extension":".py","codemirror_mode":{"name":"ipython","version":3},"name":"python","mimetype":"text/x-python"}},"nbformat_minor":4,"nbformat":4,"cells":[{"cell_type":"code","source":"!cp -r ../input/for-ptgbr/* ./","metadata":{"execution":{"iopub.status.busy":"2022-04-12T23:06:49.505973Z","iopub.execute_input":"2022-04-12T23:06:49.506712Z","iopub.status.idle":"2022-04-12T23:06:50.282995Z","shell.execute_reply.started":"2022-04-12T23:06:49.506615Z","shell.execute_reply":"2022-04-12T23:06:50.282157Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"from ssd_dssd_pytorch.ssd import TrainModel\nfrom ssd_dssd_pytorch.gbr import get_dataloader","metadata":{"execution":{"iopub.status.busy":"2022-04-12T23:06:50.286833Z","iopub.execute_input":"2022-04-12T23:06:50.287073Z","iopub.status.idle":"2022-04-12T23:06:54.186854Z","shell.execute_reply.started":"2022-04-12T23:06:50.287043Z","shell.execute_reply":"2022-04-12T23:06:54.186131Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"train_dataloader, val_dataloader = get_dataloader(\"../input/tensorflow-great-barrier-reef\")","metadata":{"execution":{"iopub.status.busy":"2022-04-12T23:06:54.188347Z","iopub.execute_input":"2022-04-12T23:06:54.188616Z","iopub.status.idle":"2022-04-12T23:06:54.473614Z","shell.execute_reply.started":"2022-04-12T23:06:54.188576Z","shell.execute_reply":"2022-04-12T23:06:54.472905Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"trainer = TrainModel(\n    train_dl=train_dataloader, val_dl=val_dataloader,\n    cfg_path=\"./ssd_dssd_pytorch/config.yaml\",\n    hyp_path=\"./ssd_dssd_pytorch/hyp.yaml\",\n    model=\"dssd\"\n)","metadata":{"execution":{"iopub.status.busy":"2022-04-12T23:06:54.475461Z","iopub.execute_input":"2022-04-12T23:06:54.475693Z","iopub.status.idle":"2022-04-12T23:07:04.581294Z","shell.execute_reply.started":"2022-04-12T23:06:54.475660Z","shell.execute_reply":"2022-04-12T23:07:04.580478Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"trainer.train()","metadata":{"execution":{"iopub.status.busy":"2022-04-12T23:07:04.582813Z","iopub.execute_input":"2022-04-12T23:07:04.583215Z","iopub.status.idle":"2022-04-12T23:24:58.764882Z","shell.execute_reply.started":"2022-04-12T23:07:04.583180Z","shell.execute_reply":"2022-04-12T23:24:58.763595Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"trainer.plot_logs()","metadata":{"execution":{"iopub.status.busy":"2022-04-12T23:24:58.766173Z","iopub.status.idle":"2022-04-12T23:24:58.766584Z","shell.execute_reply.started":"2022-04-12T23:24:58.766365Z","shell.execute_reply":"2022-04-12T23:24:58.766387Z"},"trusted":true},"execution_count":null,"outputs":[]}]}