How to freeze embedding layers

[lintangsutawika]

Hi,

I'm wondering how I might freeze token embedding layers in Unitransformer implementations. All references online seem to point to keras and not implementations with mesh.

mesh/mesh_tensorflow/transformer/transformer.py

Line 697 in 52a2332

class Unitransformer(object):

Thank you