ml4floods icon indicating copy to clipboard operation
ml4floods copied to clipboard

Published 20 hours ago • verified publisher icon spaceml-org

[Models] Model training WorldFloods

Open gonzmg88 opened this issue 4 years ago • 0 comments

  • [ ] Implement training pipeline in pytorch lightning porting code from https://gitlab.com/frontierdevelopmentlab/disaster-prevention/cubesatfloods
  • [ ] Make it flexible for trying different loss functions
  • [ ] Add metrics to W&B logger
  • [ ] Plot some image results at the end of each epoch

Nice to have:

  • Weights and Biases integration
  • Look into dvc and sacred (open-sourced experiment management systems)

gonzmg88 avatar Feb 16 '21 16:02 gonzmg88