Added capability to specify layer configuration in .train