diff --git a/src/wandb_allennlp/training/callbacks/log_to_wandb.py b/src/wandb_allennlp/training/callbacks/log_to_wandb.py
index 9c8338c..666b85a 100644
--- a/src/wandb_allennlp/training/callbacks/log_to_wandb.py
+++ b/src/wandb_allennlp/training/callbacks/log_to_wandb.py
@@ -196,6 +196,7 @@ def __init__(
             self._wandb_kwargs["config"] = flatten_dict(
                 self._wandb_kwargs["config"]
             )
+            self._wandb_kwargs["config"]["serialization_dir"] = serialization_dir
 
     def on_start(
         self,