diff --git a/Dockerfile b/Dockerfile
index 62cddf1..fa32d92 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -21,9 +21,9 @@ RUN apt-get update && \
 
 # Install PyTorch
 RUN pip3 install --no-cache-dir \
-    torch==2.2.2 \
-    torchvision==0.17.2 \
-    torchaudio==2.2.2
+    torch==2.3.0 \
+    torchvision==0.18.0 \
+    torchaudio==2.3.0
 
 # Install requirements
 COPY ./ ./
diff --git a/examples/cifar_advanced.py b/examples/cifar_advanced.py
index f514de2..7d69366 100644
--- a/examples/cifar_advanced.py
+++ b/examples/cifar_advanced.py
@@ -230,7 +230,7 @@ def train_step(self, batch, state) -> dict:
     if distributed:
         @argus.callbacks.on_epoch_complete
         def schedule_sampler(state):
-            state.data_loader.sampler.set_epoch(state.epoch + 1)
+            train_loader.sampler.set_epoch(state.epoch + 1)
         callbacks += [schedule_sampler]
 
     model.fit(train_loader,
diff --git a/pyproject.toml b/pyproject.toml
index 30c38d0..872fc18 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -35,15 +35,15 @@ dynamic = ["version"]
 tests = [
     "pytest==8.1.1",
     "pytest-cov==5.0.0",
-    "ruff==0.3.5",
+    "ruff==0.4.1",
 ]
 docs = [
-    "sphinx==7.2.6",
+    "sphinx==7.3.7",
     "pydata-sphinx-theme==0.15.2",
 ]
 examples = [
-    "torchvision==0.17.2",
-    "notebook==7.1.2",
+    "torchvision==0.18.0",
+    "notebook==7.1.3",
     "timm==0.9.16",
 ]