[limit_n_basins mode] [opt mem] [opt runtime] add mode to allow training and validation on practically any dataset size.#246

amitmarkel wants to merge 3 commits intomaingoogle-research/flood-forecasting:mainfrom

amarkel-limit_n_basins__traingoogle-research/flood-forecasting:amarkel-limit_n_basins__trainCopy head branch name to clipboard