f9c3511ae53fbc2c9fc69cadd22397f9ddfe2fb5,src/trainer.py,SamplingMultiTaskTrainer,_save_checkpoint,#SamplingMultiTaskTrainer#Any#Any#Any#Any#,621

Before Change

        // Don"t save embeddings here.
        // TODO: There has to be a prettier way to do this.
        keys_to_skip = []
        for key in model_state:
            if "token_embedder_words" in key:
                keys_to_skip.append(key)
        for key in keys_to_skip:
            del model_state[key]

        torch.save(model_state, model_path)

After Change

        // Don"t save embeddings here.
        // TODO: There has to be a prettier way to do this.
        keys_to_skip = [key for key in model_state if dont_save(key)]
        for key in keys_to_skip:
            del model_state[key]
        torch.save(model_state, model_path)

Instances