Automating Code Review Activities by Large-Scale Pre-training
Paper • 2203.09095 • Published
How to use waleko/codereviewer-finetuned-msg with Transformers:
# Load model directly
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("waleko/codereviewer-finetuned-msg")
model = AutoModelForSeq2SeqLM.from_pretrained("waleko/codereviewer-finetuned-msg")We fine-tuned the original CodeReviewer model on the Comment Generation dataset using a single A100 GPU, 8 CPU cores, and 20 GB of RAM. The fine-tuning process was run for 12 hours.
For further information please consider taking a look at the original CodeReviewer paper.