roberta_des_ada_128_6e4 / generate_pytorch_model.py

Saving weights and logs of step 10000

b87a39b over 4 years ago

256 Bytes

	# This script overwrites any existing PyTorch model. Generates a new one with an LM head from the pretrained Flax model.
	from transformers import RobertaForMaskedLM
	model = RobertaForMaskedLM.from_pretrained(".",from_flax=True)
	model.save_pretrained(".")