Corrected SAC get_snapshot. (rail-berkeley#113)

kaustubhsridhar · Jul 22, 2020 · ed28309 · ed28309
1 parent ff491f7
commit ed28309
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/rlkit/torch/sac/sac.py b/rlkit/torch/sac/sac.py
@@ -224,6 +224,6 @@ def get_snapshot(self):
             policy=self.policy,
             qf1=self.qf1,
             qf2=self.qf2,
-            target_qf1=self.qf1,
-            target_qf2=self.qf2,
+            target_qf1=self.target_qf1,
+            target_qf2=self.target_qf2,
         )