diff --git a/projects/implicitron_trainer/impl/experiment_config.py b/projects/implicitron_trainer/impl/experiment_config.py
index 27802162..30b36934 100644
--- a/projects/implicitron_trainer/impl/experiment_config.py
+++ b/projects/implicitron_trainer/impl/experiment_config.py
@@ -5,7 +5,7 @@
 # LICENSE file in the root directory of this source tree.
 
 from dataclasses import field
-from typing import Tuple
+from typing import Any, Dict, Tuple
 
 from omegaconf import DictConfig
 from pytorch3d.implicitron.dataset.data_source import ImplicitronDataSource
@@ -41,7 +41,7 @@ class ExperimentConfig(Configurable):
     clip_grad: float = 0.0
     camera_difficulty_bin_breaks: Tuple[float, ...] = 0.97, 0.98
 
-    hydra: dict = field(
+    hydra: Dict[str, Any] = field(
         default_factory=lambda: {
             "run": {"dir": "."},  # Make hydra not change the working dir.
             "output_subdir": None,  # disable storing the .hydra logs
diff --git a/projects/implicitron_trainer/impl/optimization.py b/projects/implicitron_trainer/impl/optimization.py
index 604445da..1c6a589c 100644
--- a/projects/implicitron_trainer/impl/optimization.py
+++ b/projects/implicitron_trainer/impl/optimization.py
@@ -25,7 +25,7 @@ def init_optimizer(
     gamma: float = 0.1,
     momentum: float = 0.9,
     betas: Tuple[float, ...] = (0.9, 0.999),
-    milestones: tuple = (),
+    milestones: Tuple[int, ...] = (),
     max_epochs: int = 1000,
 ):
     """