Spaces:
Running on A10G
Running on A10G
Commit ·
bf3e402
1
Parent(s): d051a6a
Fix: Enable unbuffered logging for real-time training monitoring
Browse files- Dockerfile +1 -0
Dockerfile
CHANGED
|
@@ -44,6 +44,7 @@ COPY . .
|
|
| 44 |
RUN pip install -e .
|
| 45 |
|
| 46 |
# Set environment variables
|
|
|
|
| 47 |
ENV MODEL_NAME="meta-llama/Llama-3.2-3B-Instruct"
|
| 48 |
ENV OUTPUT_DIR="outputs/commitguard-llama-3b-grpo"
|
| 49 |
ENV WANDB_PROJECT="commitguard"
|
|
|
|
| 44 |
RUN pip install -e .
|
| 45 |
|
| 46 |
# Set environment variables
|
| 47 |
+
ENV PYTHONUNBUFFERED=1
|
| 48 |
ENV MODEL_NAME="meta-llama/Llama-3.2-3B-Instruct"
|
| 49 |
ENV OUTPUT_DIR="outputs/commitguard-llama-3b-grpo"
|
| 50 |
ENV WANDB_PROJECT="commitguard"
|