Skip to content

Instantly share code, notes, and snippets.

@PallawiSinghal

PallawiSinghal/save_model_section.py

Created February 14, 2022 09:32
Show Gist options
  • Save PallawiSinghal/d3fb9bb18e6b185ae93899ee1ffc24a6 to your computer and use it in GitHub Desktop.
Save PallawiSinghal/d3fb9bb18e6b185ae93899ee1ffc24a6 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
save_model_section.py
print(cfg.dump())
with open("/code/detectron2/detectron2/output/custom_mask_rcnn_X_101_32x8d_FPN_3x_my_dataset.yaml", "w") as f:
f.write(cfg.dump())
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment