HW-202337458-trainer-callback

Assignment

06월 04일 Trainer Callback 구현 과제입니다.

Student

• Student ID: 202337458
• Hugging Face ID: hf-june

Selected Option

Option 1: Continuous training

Contents

• Custom TrainerCallback 구현
• IMDB 데이터셋 기준 구현
• 1차 학습 후 체크포인트 저장
• 체크포인트에서 Optimizer와 Scheduler 상태를 유지하며 이어서 학습
• 로그를 JSON/CSV로 저장
• 학습 로그를 연속된 그래프로 시각화

Files

• trainer_callback_option1_executed.ipynb
• trainer_callback.py
• callback_logs.json
• callback_logs.csv
• final_metrics.json
• continuous_loss_curve.png
• continuous_metric_curve.png
• README.md

Output Directory

/content/drive/MyDrive/trainer_callback_outputs/20260610_144840

Final Metrics

{
  "eval_loss": 0.5539242029190063,
  "eval_accuracy": 0.736,
  "eval_precision": 0.7435897435897436,
  "eval_recall": 0.7073170731707317,
  "eval_f1": 0.725,
  "eval_runtime": 243.0385,
  "eval_samples_per_second": 2.057,
  "eval_steps_per_second": 0.259,
  "epoch": 0.32
}

Note

The vertical line at step 40 means the resume point from checkpoint.

Created At

2026-06-10 15:27:50