BatchReNorm with warm up moved into code base

UoA-CARES · Dec 3, 2024 · 64ed5ba · 64ed5ba
1 parent 95763d3
commit 64ed5ba
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/cares_reinforcement_learning/networks/CrossQ/actor.py b/cares_reinforcement_learning/networks/CrossQ/actor.py
@@ -1,7 +1,7 @@
 import torch
-from batchrenorm import BatchRenorm1d
 from torch import nn
 
+from cares_reinforcement_learning.util.batchrenorm import BatchRenorm1d
 from cares_reinforcement_learning.util.common import SquashedNormal
 from cares_reinforcement_learning.util.configurations import CrossQConfig
 

diff --git a/cares_reinforcement_learning/networks/CrossQ/critic.py b/cares_reinforcement_learning/networks/CrossQ/critic.py
@@ -1,7 +1,7 @@
 import torch
-from batchrenorm import BatchRenorm1d
 from torch import nn
 
+from cares_reinforcement_learning.util.batchrenorm import BatchRenorm1d
 from cares_reinforcement_learning.util.configurations import CrossQConfig