File size: 761 Bytes
f211247 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 | #!/bin/bash
#SBATCH --job-name=p5_seq2seq_ft
#SBATCH --partition=akya-cuda
#SBATCH --nodes=1
#SBATCH --gres=gpu:4
#SBATCH --cpus-per-task=40
#SBATCH --mem=192G
#SBATCH --time=24:00:00
#SBATCH --output=/arf/scratch/stakan/hitit-proje/logs/p5/08_seq2seq_ft_%j.log
#SBATCH --error=/arf/scratch/stakan/hitit-proje/logs/p5/08_seq2seq_ft_%j.err
source /arf/scratch/stakan/hitit-proje/hitit_ocr/scripts/pipeline_p5/_common.sh
python3 hitit_ocr/src/seq2seq/train_image.py \
--config hitit_ocr/configs/seq2seq_hitit.yaml \
--phase finetune \
--encoder hitit_ocr/runs/p5/ssl_dinov3/checkpoint.pt \
--decoder hitit_ocr/runs/p5/seq2seq_text_pretrain/checkpoint.pt \
--data datasets/processed/seq2seq_pairs.jsonl \
--output hitit_ocr/runs/p5/seq2seq_image_ft/
|