search

NUSTM / VLP-MABSA

107 stars 10 forks source link

Where can I modify batch size and learning rate? #22

Closed Jackson8787 closed 1 year ago

Jackson8787 commented 1 year ago

I ran out the vram during the pre-training task .Where can I modify batch size and learning rate？ 230147663-33dde6f4-afa2-420e-9bae-c5cb23846658

lyhuohuo commented 1 year ago

You can add "--lr" and "--batch_size" parameter in MVSA_pretrain.sh or just modify line 312 and line 397 in pretrain.py.