def set_num_updates(self, num_updates):
        """State from trainer to pass along to model at every update."""
        for m in self.modules():
            if hasattr(m, "set_num_updates") and m != self:
                m.set_num_updates(num_updates)
State from trainer to pass along to model at every update.
understand_icl/fairseq/fairseq/models/fairseq_model.py