diff --git a/docker/tf_1/Dockerfile b/docker/tf_1/Dockerfile
index 7408249..bc5e9f6 100644
--- a/docker/tf_1/Dockerfile
+++ b/docker/tf_1/Dockerfile
@@ -1,6 +1,9 @@
 # Base Image
 FROM tensorflow/tensorflow:1.15.2-gpu-py3
 
+RUN rm /etc/apt/sources.list.d/cuda.list
+RUN rm /etc/apt/sources.list.d/nvidia-ml.list
+
 RUN apt-get update && apt-get -q -y install git-core wget
 
 # Installing the Tensorflow Object Detection API
diff --git a/docker/tf_2/Dockerfile b/docker/tf_2/Dockerfile
index fd7b99d..4856778 100644
--- a/docker/tf_2/Dockerfile
+++ b/docker/tf_2/Dockerfile
@@ -1,6 +1,9 @@
 # Base Image
 FROM tensorflow/tensorflow:2.6.0-gpu
 
+RUN rm /etc/apt/sources.list.d/cuda.list
+RUN rm /etc/apt/sources.list.d/nvidia-ml.list
+
 RUN apt-get update && apt-get -q -y install git-core wget
 
 # Installing the Tensorflow Object Detection API
diff --git a/docker/tf_2/requirements.txt b/docker/tf_2/requirements.txt
index 5583efe..ac0b728 100644
--- a/docker/tf_2/requirements.txt
+++ b/docker/tf_2/requirements.txt
@@ -13,4 +13,5 @@ Pillow==8.4.0
 jellyfish==0.8.2
 matplotlib==3.3.4
 opencv-python-headless==4.5.3.56
-pyparsing==2.4.7
\ No newline at end of file
+pyparsing==2.4.7
+pyOpenSSL==22.0.0
\ No newline at end of file
diff --git a/src/deployment/conda_env.yml b/src/deployment/conda_env.yml
index c0f4247..2877624 100644
--- a/src/deployment/conda_env.yml
+++ b/src/deployment/conda_env.yml
@@ -11,5 +11,7 @@ dependencies:
   - tensorflow-gpu==1.15
   - numpy==1.21.5
   - itsdangerous==2.0.1
+  - Jinja2==3.0.3
+  - protobuf==3.20.0
 
 name: basic_env
\ No newline at end of file
diff --git a/src/deployment/conda_env_tf2.yml b/src/deployment/conda_env_tf2.yml
index d93e585..c7986c4 100644
--- a/src/deployment/conda_env_tf2.yml
+++ b/src/deployment/conda_env_tf2.yml
@@ -11,5 +11,7 @@ dependencies:
   - tensorflow==2.8.0
   - numpy==1.21.5
   - itsdangerous==2.0.1
+  - Jinja2==3.0.3
+  - protobuf==3.20.0
 
 name: basic_env
\ No newline at end of file
diff --git a/src/training/exp_config_sample.json b/src/training/exp_config_sample.json
index 9a677ec..732e569 100644
--- a/src/training/exp_config_sample.json
+++ b/src/training/exp_config_sample.json
@@ -9,6 +9,7 @@
     "IMAGE_TYPE" : "",
     "TRAIN_CSV" : "",
     "TEST_CSV" : "",
+    "BATCH_SIZE": 1,
     "EVAL_CONF" : 0.5,
     "RUN_PARAMS": {
         "STEPS": 1000
diff --git a/src/training/submit_training.py b/src/training/submit_training.py
index 833e12b..17c76f5 100644
--- a/src/training/submit_training.py
+++ b/src/training/submit_training.py
@@ -43,6 +43,7 @@ def main():
     img_type = exp_config['IMAGE_TYPE']
     train_csv = exp_config['TRAIN_CSV']
     test_csv = exp_config['TEST_CSV']
+    batch_size = exp_config['BATCH_SIZE']
     base_model = (exp_config['MODEL_PARAMS']
                             ['BASE_MODEL'])
     steps = (exp_config['RUN_PARAMS']
@@ -87,6 +88,7 @@ def main():
             '--image_type', img_type,
             '--train_csv', train_csv,
             '--test_csv', test_csv,
+            '--batch_size', batch_size,
             '--base_model', base_model,
             '--steps', steps,
             '--fs_nms_iou', fs_nms_iou,
@@ -109,6 +111,7 @@ def main():
             '--image_type', img_type,
             '--train_csv', train_csv,
             '--test_csv', test_csv,
+            '--batch_size', batch_size,
             '--base_model', base_model,
             '--steps', steps,
             '--eval_conf', eval_conf]