Python TorchPolicy.extra_action_out Examples

Example #1

 def extra_action_out(self, input_dict, state_batches, model_out):
     if extra_action_out_fn:
         return extra_action_out_fn(self, input_dict, state_batches,
                                    model_out)
     else:
         return TorchPolicy.extra_action_out(self, input_dict,
                                             state_batches, model_out)

Example #2

File: torch_policy_template.py Project: vrishank97/ray

 def extra_action_out(self, input_dict, state_batches, model,
                      action_dist):
     with torch.no_grad():
         if extra_action_out_fn:
             stats_dict = extra_action_out_fn(
                 self, input_dict, state_batches, model, action_dist)
         else:
             stats_dict = TorchPolicy.extra_action_out(
                 self, input_dict, state_batches, model, action_dist)
         return convert_to_non_torch_type(stats_dict)

Example #3

 def extra_action_out(self, model_out):
     if extra_action_out_fn:
         return extra_action_out_fn(self, model_out)
     else:
         return TorchPolicy.extra_action_out(self, model_out)