update `utils/gen_saved_model.py` for binary NCLASSES=2 model

Doodleverse / segmentation_gym

A neural gym for training deep learning models to carry out geoscientific image segmentation. Works best with labels generated using https://github.com/Doodleverse/dash_doodler

MIT License

45 stars 11 forks source link

update `utils/gen_saved_model.py` for binary NCLASSES=2 model #109

Closed dbuscombe-usgs closed 2 years ago

dbuscombe-usgs commented 2 years ago

minor update involving specifying NCLASSES for the new convention of NCLASSES>=2

dbuscombe-usgs commented 2 years ago

Fixed with https://github.com/Doodleverse/segmentation_gym/commit/0234f43796e52e87550da9c7fd49c68d594a2864