Views
No views yet
1from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
2
3model_checkpoint = "LazarusNLP/IndoNanoT5-base"
4
5tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
6model = AutoModelForSeq2SeqLM.from_pretrained(model_checkpoint)total_steps: 65536input_length: 512batch_size: 128grad_acc: 1base_lr: 5e-3optimizer: AdamWScaled with betas=(0.9,0.999) and epsilon=1e-08weight_decay: 0.0lr_scheduler: cosinewarmup_steps: 10000final_cosine: 1e-5grad_clip: 1.0precision: bf16