diff --git a/Models/ImageClassification/WideResNet.swift b/Models/ImageClassification/WideResNet.swift
index 2852eb7c33b..ffd89c83b07 100644
--- a/Models/ImageClassification/WideResNet.swift
+++ b/Models/ImageClassification/WideResNet.swift
@@ -26,8 +26,8 @@ public struct BatchNormConv2DBlock: Layer {
     public var norm2: BatchNorm<Float>
     public var conv2: Conv2D<Float>
     public var shortcut: Conv2D<Float>
-    let isExpansion: Bool
-    let dropout: Dropout<Float> = Dropout(probability: 0.3)
+    @noDerivative let isExpansion: Bool
+    @noDerivative let dropout: Dropout<Float> = Dropout(probability: 0.3)
 
     public init(
         featureCounts: (Int, Int),