mirror of
https://github.com/wassname/kair_algorithms_draft.git
synced 2026-10-29 13:10:50 +08:00
Add soft actor critic (#12)
* Add soft actor critic * Delete unnecessary examples
This commit is contained in:
1 parent
d2b670015c
commit
4a19053d25
6 files changed
+589
-17
No files matched your search
@@ -12,6 +12,8 @@ import numpy as np
|
||||
import torch
|
||||
import torch.nn as nn
|
||||
|
||||
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
||||
|
||||
|
||||
def identity(x: torch.Tensor) -> torch.Tensor:
|
||||
"""Return input without any change."""
|
||||
|
||||
Reference in new issue
Block a user