diff --git a/examples/ImageNet/Project.toml b/examples/ImageNet/Project.toml
index f5dd3601f..6d2d99e1d 100644
--- a/examples/ImageNet/Project.toml
+++ b/examples/ImageNet/Project.toml
@@ -1,4 +1,5 @@
 [deps]
+AMDGPU = "21141c5a-9bdb-4563-92ae-f87d6854732e"
 Augmentor = "02898b10-1f73-11ea-317c-6393d7073e15"
 Boltz = "4544d5e4-abc5-4dea-817f-29e4c205d9c8"
 Configurations = "5218b696-f38b-4ac9-8b61-a12ec717816d"
@@ -11,7 +12,6 @@ Images = "916415d5-f1e6-5110-898d-aaa5f9f070e0"
 JLD2 = "033835bb-8acc-5ee8-8aae-3f567f8a3819"
 JpegTurbo = "b835a17e-a41a-41e7-81f0-2f016b05efe0"
 Lux = "b2108857-7c20-44ae-9111-449ecde12c47"
-AMDGPU = "21141c5a-9bdb-4563-92ae-f87d6854732e"
 LuxCUDA = "d0bbae9a-e099-4d5b-a835-1c6931763bda"
 MLUtils = "f1d291b0-491e-4a28-83b9-f70985020b54"
 MPI = "da04e1cc-30fd-572f-bb4f-1f8673147195"
@@ -36,7 +36,7 @@ FileIO = "1.16"
 Format = "1.3"
 Functors = "0.4"
 Images = "0.26"
-JLD2 = "0.4.46"
+JLD2 = "0.4.46, 0.5"
 JpegTurbo = "0.1"
 Lux = "1"
 LuxCUDA = "0.3"