From 967e7468aef53cb8d2ad75009bbcd694c89a913f Mon Sep 17 00:00:00 2001 From: Xu Zhao Date: Thu, 20 Jun 2024 23:38:55 -0400 Subject: [PATCH] Fix sac --- torchbenchmark/models/soft_actor_critic/sac.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchbenchmark/models/soft_actor_critic/sac.py b/torchbenchmark/models/soft_actor_critic/sac.py index dd590a8d02..c67da4fca5 100644 --- a/torchbenchmark/models/soft_actor_critic/sac.py +++ b/torchbenchmark/models/soft_actor_critic/sac.py @@ -3,7 +3,7 @@ import numpy as np import torch -from . import envs, nets, replay, utils +from . import envs, nets, replay, sac_utils class SACAgent: