dodox/metastable-baselines2
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fixed bug in SAC

Browse Source
This commit is contained in:
Dominik Moritz Roth 2023-08-22 00:17:09 +02:00
parent 1f0748e49f
commit a873131704
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
sbBrix/sac/sac.py
View File

@ -8,7 +8,7 @@ from torch.nn import functional as F
from stable_baselines3.common.buffers import ReplayBuffer
from stable_baselines3.common.noise import ActionNoise
# from stable_baselines3.common.off_policy_algorithm import OffPolicyAlgorithm
from common.off_policy_algorithm import BetterOffPolicyAlgorithm
from ..common.off_policy_algorithm import BetterOffPolicyAlgorithm
from stable_baselines3.common.policies import BasePolicy
from stable_baselines3.common.type_aliases import GymEnv, MaybeCallback, Schedule
from stable_baselines3.common.utils import get_parameters_by_name, polyak_update