mapbox / robosat

Semantic segmentation on aerial and satellite imagery. Extracts features such as: buildings, parking lots, roads, water, clouds
MIT License
2.02k stars 382 forks source link

Move asserts to start of the training command #141

Closed NickStallman closed 5 years ago

NickStallman commented 5 years ago

I just wasted ~12 hours because ./rs train ran for a single epoch and then right at the end it asserted in def validate on: assert num_samples > 0, "dataset contains training images and labels"

This is of course not ideal at all, it would be best for things like that to be checked before any training work is done.