Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
mamengyiyi authored Apr 21, 2022
1 parent 33d6506 commit f470969
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion offline-rl-algorithms/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -182,7 +182,7 @@ get_dataset(args.dataset)
```
### Modify Algorithm
Assuming you're modifying algorithm based on SAC:
* Create two python file, name them as *YourSAC.py* and *YourSACImpl.py*.
* Create two python file, name them as *YourSAC.py* and *YourSACImpl.py*. 其中*YourSACImpl.py*中指定的*YourSACImpl* class继承*SACImpl*.
* Modify your algo in *YourSACImpl.py* by overloading *compute_critic_loss/compute_actor_loss/other* functions.
```
def compute_critic_loss(self, batch: TorchMiniBatch, q_tpn: torch.Tensor) -> torch.Tensor:
Expand Down

0 comments on commit f470969

Please sign in to comment.