def __call__(self, s, visual_s, *args, cell_state=(None,), **kwargs):
    (feat, cell_state) = self.representation_net(s, visual_s, cell_state)
    output = self.value_net(feat, *args, **kwargs)
    output2 = self.value_net2(feat, *args, **kwargs)
    return (output, output2, cell_state)
def __call__(self, s, visual_s, *args, cell_state=(None,), **kwargs):
    (feat, cell_state) = self.representation_net(s, visual_s, cell_state)
    output = self.value_net(feat, *args, **kwargs)
    output2 = self.value_net2(feat, *args, **kwargs)
    return (output, output2, cell_state)