search

spaceml-org / ml4floods

An ecosystem of data, models and code pipelines to tackle flooding with ML
https://spaceml-org.github.io/ml4floods/
GNU Lesser General Public License v3.0
133 stars 40 forks source link

Worldfloodsv1 training #65

Closed gonzmg88 closed 3 years ago

gonzmg88 commented 3 years ago

We fixed some bad bugs from yesterday's release:

  1. shuffle in training
  2. dataloader with multiple workers
  3. weighting the loss by the inverse of the class frequency
  4. Nicer logging of losses and images in W&B to distinguish between train and val metrics