Skip to content

Commit bfc3d16

Browse files
committed
fix FastSAC evaluate_policy missing set_is_evaluating call
1 parent 8b789d9 commit bfc3d16

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/holosoma/holosoma/agents/fast_sac/fast_sac_agent.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1000,6 +1000,7 @@ def export(self, onnx_file_path: str) -> None:
10001000

10011001
@torch.no_grad()
10021002
def evaluate_policy(self, max_eval_steps: int | None = None):
1003+
self.env.set_is_evaluating()
10031004
obs = self.env.reset()
10041005

10051006
for _ in itertools.islice(itertools.count(), max_eval_steps):

0 commit comments

Comments
 (0)