Fix #342

* Fix lots of trainer_config_helpers bug, and complete unittest for `layers.py`