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