taavipall/prokbert repository overview

git clone https://github.com/nbrg-ppcu/prokbert
cd prokbert

docker run --rm -v $(pwd)/:/app/ taavipall/prokbert \
    python /app/examples/finetuning.py \
    --model_name neuralbioinfo/prokbert-mini \
    --ftmodel mini_promoter \
    --model_outputpath finetuning_outputs \
    --num_train_epochs 1 \
    --per_device_train_batch_size 128

docker run --rm -p 8888:8888 -v $(pwd)/:/app/ taavipall/prokbert \
    jupyter notebook \
    --ip 0.0.0.0 \
    --no-browser \
    --allow-root