Improve support for dropout in DIPS
Summary
This MR introduces the following changes
- Adds the option to specify a dropout rate for each layer individually (in DIPS)
- Adds a small helper function that reads the rates from a config dict and compares the size to the number of layers