Add state, observation and image dimensions in global config and use it everywhere

[Refinath]

Add state, observation and image dimensions in global config and use it every where and replace hydra based dimensions