Fixed bug in SAC
This commit is contained in:
parent
1f0748e49f
commit
a873131704
@ -8,7 +8,7 @@ from torch.nn import functional as F
|
|||||||
from stable_baselines3.common.buffers import ReplayBuffer
|
from stable_baselines3.common.buffers import ReplayBuffer
|
||||||
from stable_baselines3.common.noise import ActionNoise
|
from stable_baselines3.common.noise import ActionNoise
|
||||||
# from stable_baselines3.common.off_policy_algorithm import OffPolicyAlgorithm
|
# 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.policies import BasePolicy
|
||||||
from stable_baselines3.common.type_aliases import GymEnv, MaybeCallback, Schedule
|
from stable_baselines3.common.type_aliases import GymEnv, MaybeCallback, Schedule
|
||||||
from stable_baselines3.common.utils import get_parameters_by_name, polyak_update
|
from stable_baselines3.common.utils import get_parameters_by_name, polyak_update
|
||||||
|
Loading…
Reference in New Issue
Block a user