def instantiate(self, **extra_args):
        """ Instantiate the model """
        input_block = self.input_block.instantiate()
        backbone = self.backbone.instantiate(**extra_args)

        return StochasticPolicyModel(input_block, backbone, extra_args['action_space'])
def instantiate ( self , ** extra_args ) : input_block = self . input_block . instantiate ( ) backbone = self . backbone . instantiate ( ** extra_args ) return StochasticPolicyModel ( input_block , backbone , extra_args [ 'action_space' ] )
Instantiate the model