Skip to content

GitLab

Unverified Commit d7dcfa89 authored by Pingzhi Li's avatar Pingzhi Li Committed by GitHub
Browse files

Remove an unexpected argument for FlaxResNetBasicLayerCollection (#27272) 

Remove unexpected argument for FlaxResNetBasicLayerCollection
parent eef7ea98
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
src/transformers/models/resnet/modeling_flax_resnet.py
View file @ d7dcfa89
...@@ -215,7 +215,6 @@ class FlaxResNetBasicLayer(nn.Module): ...@@ -215,7 +215,6 @@ class FlaxResNetBasicLayer(nn.Module):
self.layer = FlaxResNetBasicLayerCollection( self.layer = FlaxResNetBasicLayerCollection(
out_channels=self.out_channels, out_channels=self.out_channels,
stride=self.stride, stride=self.stride,
activation=self.activation,
dtype=self.dtype, dtype=self.dtype,
) )
self.activation_func = ACT2FN[self.activation] self.activation_func = ACT2FN[self.activation]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment