Running APEX-DQN with custom model

Hi,

I am trying to use APEX-DQN algorithm to train my agent which uses a custom model. My custom model works well when using DQN algorithm but causes an error with APEX+DQN giving me the following error: KeyError: ‘fcnet_hiddens’

Can someone help me out on how to solve this problem? When I still use my custom model, why is rllib trying to fetch default config?

Never mind, figured the error!