Merge pull request #13 from UoA-CARES/docker

Docker

.gitignore

@@ -3,7 +3,6 @@ install
 log
 data
 notes
-models
 training_logs
 checkpoint
 cares_reinforcement_learning

Dockerfile

@@ -29,6 +29,4 @@ RUN cd cares_reinforcement_learning && \
 RUN echo 'source /opt/ros/humble/setup.bash' >> ~/.bashrc && \
     echo 'source install/setup.bash' >> ~/.bashrc
 
-WORKDIR /ws
-
-RUN pip install -r requirements.txt
+WORKDIR /ws

dockercommand.txt

@@ -0,0 +1 @@
+docker run --rm -it --network host --gpus all -e DISPLAY -e GZ_PARTITION=10 -v "$PWD/data:/ws/data" -v "$PWD/models:/ws/models" -v "$PWD/figures:/ws/figures" autonomous_f1tenth:latest bash

run.sh

@@ -12,7 +12,6 @@ xhost +
 docker run --rm -it \
     --network host \
     --gpus all \
-    --privileged \
     -e NVIDIA_VISIBLE_DEVICES=all \
     -e NVIDIA_DRIVER_COMPABILITIES=all \
     -e QT_X11_NO_MITSHM=1 \
@@ -24,7 +23,6 @@ docker run --rm -it \
     -v "$PWD/models:/ws/models" \
     -v "$PWD/figures:/ws/figures" \
     -v "$PWD/src:/ws/src" \
-    -v "/tmp/.X11-unix/:/tmp/.X11-unix" \
     autonomous_f1tenth:latest \
     bash
 

src/reinforcement_learning/reinforcement_learning/DataManager.py

@@ -27,7 +27,7 @@ def post(self, y_point):
         self.y_data.append(y_point)
 
         if self.plot_num % self.checkpoint_freq == 0:
-            self.save_csv(f'{self.name}-checkpoint-{len(self.x_data)}.csv')
+            self.save_csv(f'{self.name}-checkpoint.csv')
 
     def save_csv(self, file_name=str(uuid.uuid4().hex)):
         dir_exists = os.path.exists("data")