mirror of
https://github.com/huggingface/transformers.git
synced 2025-07-31 02:02:21 +06:00
Remove an unexpected argument for FlaxResNetBasicLayerCollection (#27272)
Remove unexpected argument for FlaxResNetBasicLayerCollection
This commit is contained in:
parent
eef7ea98c3
commit
d7dcfa8917
@ -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]
|
||||
|
Loading…
Reference in New Issue
Block a user