Remove an unexpected argument for FlaxResNetBasicLayerCollection (#27272)

Remove unexpected argument for FlaxResNetBasicLayerCollection

Remove an unexpected argument for FlaxResNetBasicLayerCollection (#27272)
Remove unexpected argument for FlaxResNetBasicLayerCollection
d7dcfa89 · Pingzhi Li · GitHub · eef7ea98 · d7dcfa89
Unverified Commit d7dcfa89 authored Nov 06, 2023 by Pingzhi Li Committed by GitHub Nov 06, 2023
Show whitespace changes
Inline Side-by-side

Showing with 0 additions and 1 deletion

src/transformers/models/resnet/modeling_flax_resnet.py src/transformers/models/resnet/modeling_flax_resnet.py +0 -1

No files found.
--- a/src/transformers/models/resnet/modeling_flax_resnet.py
+++ b/src/transformers/models/resnet/modeling_flax_resnet.py
@@ -215,7 +215,6 @@ class FlaxResNetBasicLayer(nn.Module):
        self.layer = FlaxResNetBasicLayerCollection(
            out_channels=self.out_channels,
            stride=self.stride,
-            activation=self.activation,
            dtype=self.dtype,
        )
        self.activation_func = ACT2FN[self.activation]