search

aig-upf / Partition-HRL

1 stars 0 forks source link

question for Lorenzo #5

Closed DamienAllonsius closed 5 years ago

DamienAllonsius commented 5 years ago

Is it normal that the shape of value below is : value = [[number]] ? I have tu return value[0][0] to get a number and not a list...

    def get_value(self, state):

        value = self.main_model_nn.prediction_critic([state])
        return value[0][0]

(see file agent_a2c.py) thanks !