wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[pre-commit.ci] auto fixes from pre-commit.com hooks 

for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2025-03-12 10:16:54 +00:00
parent 5081c145dc
commit 41219fe81e
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
lerobot/common/policies/sac/modeling_sac.py
View File

@ -231,7 +231,7 @@ class SACPolicy(
# Load or create config # Load or create config
if config_file and os.path.exists(config_file): if config_file and os.path.exists(config_file):
# Load config from file # Load config from file
with open(config_file, "r") as f: with open(config_file) as f:
config_dict = json.load(f) config_dict = json.load(f)
config = SACConfig(**config_dict) config = SACConfig(**config_dict)
else: else:
@ -690,9 +690,9 @@ class Policy(nn.Module):
# Compute standard deviations # Compute standard deviations
if self.fixed_std is None: if self.fixed_std is None:
log_std = self.std_layer(outputs) log_std = self.std_layer(outputs)
assert not torch.isnan(log_std).any(), ( assert not torch.isnan(
"[ERROR] log_std became NaN after std_layer!" log_std
) ).any(), "[ERROR] log_std became NaN after std_layer!"
if self.use_tanh_squash: if self.use_tanh_squash:
log_std = torch.tanh(log_std) log_std = torch.tanh(log_std)