mirror of
https://github.com/wassname/ray.git
synced 2026-10-04 13:00:39 +08:00
1 parent
fb40787603
commit
fe5bd09b46
2 files changed
+6
-4
No files matched your search
@@ -52,10 +52,10 @@ Policies can be implemented using `any framework <https://github.com/ray-project
|
||||
return -tf.reduce_mean(
|
||||
action_dist.logp(train_batch["actions"]) * train_batch["rewards"])
|
||||
|
||||
# <class 'ray.rllib.policy.tf_policy_template.MyTFPolicy'>
|
||||
MyTFPolicy = build_tf_policy(
|
||||
name="MyTFPolicy",
|
||||
loss_fn=policy_gradient_loss)
|
||||
# <class 'ray.rllib.policy.tf_policy_template.MyTFPolicy'>
|
||||
MyTFPolicy = build_tf_policy(
|
||||
name="MyTFPolicy",
|
||||
loss_fn=policy_gradient_loss)
|
||||
|
||||
Sample Batches
|
||||
~~~~~~~~~~~~~~
|
||||
|
||||
Reference in new issue
Block a user