demo_training not fully executable

[andrzejnovak]

_Config seems to be missing a param

and the plotter does not match the network output shape

[phinate]

Thanks -- good catch! This is me forgetting to turn something off I was experimenting with.

I'll push something by the end of the day, but for now if you want to use it, just change bounds in fit to m.config.suggested_bounds().

[andrzejnovak]

👍 cheers

You might want to consider adding papermill to test your notebooks before merging :)