pino-bigbird-roberta-base / create_pt_model.py
dat
Saving weights and logs of step 240000
dc01ff5
from transformers import BigBirdForMaskedLM
import torch
pt_model = BigBirdForMaskedLM.from_pretrained(".", from_flax=True)
pt_model.save_pretrained(".",save_config=False)