add more comments

Author: pan-x-c

scripts/docker/Dockerfile

@@ -1,3 +1,4 @@
+# This Dockerfile sets up a Trinity-RFT environment with minimal support.
 # Build and run the docker image with the following command:
 #
 # cd <Trinity-RFT root dir>

scripts/docker/Dockerfile.megatron

@@ -1,3 +1,4 @@
+# This Dockerfile sets up a Trinity-RFT environment with Megatron-LM support.
 # Build and run the docker image with the following command:
 #
 # cd <Trinity-RFT root dir>
@@ -36,7 +37,7 @@ RUN pip install --upgrade pip \
         --resume-retries 20 git+https://github.com/NVIDIA/apex.git
 
 # Set Env variables
-
+j
 # WANDB
 # ENV WANDB_API_KEY=
 # ENV WANDB_BASE_URL=

scripts/docker/Dockerfile.uv

@@ -1,9 +1,13 @@
+# This Dockerfile sets up a Trinity-RFT environment with Megatron support using uv.
 # Build and run the docker image with the following command:
 #
 # cd <Trinity-RFT root dir>
 # docker build -f scripts/docker/Dockerfile.uv -t trinity-rft:latest .
 # docker run -it --gpus all --shm-size="64g" --rm -v $PWD:/workspace -v <root_path_of_data_and_checkpoints>:/data trinity-rft:latest
-
+#
+# Note:
+# 1. This Dockerfile uses 'uv' to create a virtual environment for better package management. If you want a simpler setup without 'uv', please refer to `scripts/docker/Dockerfile`.
+# 2. Make sure to use `uv pip` to install packages within the virtual environment.
 
 FROM nvcr.io/nvidia/cuda:12.8.1-cudnn-devel-ubuntu22.04