fix missing import in dockerfile

minor comments
zauberzeug · Aug 30, 2023 · 95df480 · 95df480
1 parent 05f1900
commit 95df480
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/trainer/Dockerfile b/trainer/Dockerfile
@@ -19,7 +19,7 @@ RUN apt update && \
     rm -rf /var/lib/apt/lists/*
 RUN python3 -m pip install --upgrade pip
 
-RUN python3 -m pip install seaborn thop coremltools onnx gsutil notebook wandb>=0.12.2 opencv-python==4.5.5.64 torchinfo python-dotenv
+RUN python3 -m pip install seaborn thop coremltools onnx gsutil notebook wandb>=0.12.2 opencv-python==4.5.5.64 torchinfo python-dotenv httptools
 
 WORKDIR /
 

diff --git a/trainer/background_detect.py b/trainer/background_detect.py
@@ -17,6 +17,8 @@
 
 logging.basicConfig(level=logging.INFO)
 
+# NOTE: This feature is not yet ready to use and is not using the newest node lib
+
 
 @dataclass
 class Project:

diff --git a/trainer/batch_size_calculation.py b/trainer/batch_size_calculation.py
@@ -54,7 +54,7 @@ def _calc_batch_size(
     free_mem = trainer_utils.get_free_memory_mb()
     fraction = 0.95
     free_mem *= fraction
-    logging.info(f'We use only {fraction *100}% of the free memory ({free_mem})')
+    logging.info(f'{fraction:.0%} of free memory ({free_mem}) in use')
 
     device = torch.device('cuda', 0)
     torch.cuda.empty_cache()