Remove an unexpected argument for FlaxResNetBasicLayerCollection (#27272)

Remove unexpected argument for FlaxResNetBasicLayerCollection
This commit is contained in:
Pingzhi Li 2023-11-06 20:16:03 +08:00 committed by GitHub
parent eef7ea98c3
commit d7dcfa8917
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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]