Sure, I have a model that has nn.Embedding, the value of which I pass explicitly. My data changes from time to time, so does the embedding matrix and its shape. I want this tensor to be not part of the checkpoint.