mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-23 12:40:08 +00:00
fix docstring
This commit is contained in:
@@ -101,7 +101,7 @@ class SACAlgorithm(RLAlgorithm):
|
|||||||
self.policy.discrete_critic = self.discrete_critic
|
self.policy.discrete_critic = self.discrete_critic
|
||||||
|
|
||||||
def _init_discrete_critics(self, encoder: SACObservationEncoder) -> tuple[DiscreteCritic, DiscreteCritic]:
|
def _init_discrete_critics(self, encoder: SACObservationEncoder) -> tuple[DiscreteCritic, DiscreteCritic]:
|
||||||
"""Build discrete discrete critic ensemble and target networks."""
|
"""Build discrete critic ensemble and target networks."""
|
||||||
discrete_critic = DiscreteCritic(
|
discrete_critic = DiscreteCritic(
|
||||||
encoder=encoder,
|
encoder=encoder,
|
||||||
input_dim=encoder.output_dim,
|
input_dim=encoder.output_dim,
|
||||||
|
|||||||
Reference in New Issue
Block a user