Closed mykolaskrynnyk closed 5 months ago
@mykolaskrynnyk should I wait for the kwargs
doc update or just pull in tie_weights
docs for now and we do that on separate PR?
@mattdangerw , I've just pushed the changes that we discussed earlier. For ReversibleEmbedding
, I referenced keras.layers.Embedding
in the docstring as that is the class it inherits from directly.
I think we can make similar changes to preprocessing
layers. However, I can see that RandomSwap
and RandomDeletion
expect a specific dtype
by default, so we might need to change the wording slightly. At any rate, that will be a different pull request. Cheers.
Thanks very much for the contribution!
Improves the documentation in
layers/modeling
byTokenAndPositionEmbedding
.name
argument inTransformerDecoder
andTransformerEncoder
layers with that inFNetEncoder
layer as well as their own docstrings.