# 성능지표 #3(메타 데이터 추출 F1) 학습·평가 전용 이미지.
# KLUE NER 토큰 분류는 GPU 를 쓰므로 표절 평가용 py39 이미지와 분리한다.
FROM pytorch/pytorch:2.3.1-cuda12.1-cudnn8-runtime
WORKDIR /app
RUN pip install --no-cache-dir \
    "transformers==4.44.2" "datasets==2.21.0" "seqeval==1.2.2" \
    "accelerate==0.34.2" "numpy<2"
COPY scripts/train_klue_ner.py ./scripts/
CMD ["python", "scripts/train_klue_ner.py"]