Change GPTAdamWFactory to AdamWFactory in DT reproduction

takuseno · Oct 9, 2023 · f53fd3b · f53fd3b
1 parent 589c1f2
commit f53fd3b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/reproductions/offline/decision_transformer.py b/reproductions/offline/decision_transformer.py
@@ -28,7 +28,7 @@ def main() -> None:
     dt = d3rlpy.algos.DecisionTransformerConfig(
         batch_size=64,
         learning_rate=1e-4,
-        optim_factory=d3rlpy.models.GPTAdamWFactory(weight_decay=1e-4),
+        optim_factory=d3rlpy.models.AdamWFactory(weight_decay=1e-4),
         encoder_factory=d3rlpy.models.VectorEncoderFactory(
             [128],
             exclude_last_activation=True,