Fix permission: use /tmp/training as workdir with proper permissions

Dockerfile

@@ -1,6 +1,7 @@
 FROM pytorch/pytorch:2.2.0-cuda12.1-cudnn8-devel
 
-WORKDIR /app
+# 書き込み可能なワークディレクトリ
+WORKDIR /tmp/training
 
 # 基本パッケージ
 RUN apt-get update && apt-get install -y \
@@ -21,8 +22,9 @@ ENV HF_TOKEN=""
 ENV HF_HOME=/tmp/hf_cache
 ENV TRANSFORMERS_CACHE=/tmp/hf_cache
 
-# キャッシュディレクトリ作成
-RUN mkdir -p /tmp/hf_cache && chmod 777 /tmp/hf_cache
+# ディレクトリ作成
+RUN mkdir -p /tmp/hf_cache /tmp/training/checkpoints /tmp/training/output && \
+    chmod -R 777 /tmp/hf_cache /tmp/training
 
 # シングルGPU学習 (L40S 48GB)
 CMD ["python", "train.py"]