mirror of
https://github.com/wassname/kair_algorithms_draft.git
synced 2026-09-09 11:25:10 +08:00
Add soft actor critic (#12)
* Add soft actor critic * Delete unnecessary examples
This commit is contained in:
@@ -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