From 256bf72f8e32a66ddd5749a5bec155a93eb7020f Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 10 Sep 2019 12:40:59 +0200 Subject: [PATCH] updates --- Dockerfile | 2 +- utils/utils.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/Dockerfile b/Dockerfile index d4cdee31..9e14453a 100644 --- a/Dockerfile +++ b/Dockerfile @@ -2,11 +2,11 @@ FROM nvcr.io/nvidia/pytorch:19.08-py3 # Install dependencies (pip or conda) +RUN pip install -U gsutil # RUN pip install -U -r requirements.txt # RUN conda update -n base -c defaults conda # RUN conda install -y -c anaconda future numpy opencv matplotlib tqdm pillow # RUN conda install -y -c conda-forge scikit-image tensorboard pycocotools -# conda install pytorch torchvision -c pytorch # Install OpenCV with Gstreamer support #WORKDIR /usr/src diff --git a/utils/utils.py b/utils/utils.py index 87c0b567..a28ed5b3 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -684,7 +684,7 @@ def print_mutation(hyp, results, bucket=''): with open('evolve.txt', 'a') as f: # append result f.write(c + b + '\n') x = np.unique(np.loadtxt('evolve.txt', ndmin=2), axis=0) # load unique rows - np.savetxt('evolve.txt', x[np.argsort(-fitness(x))], '%10.3g') # save sort by fitness + np.savetxt('evolve.txt', x[np.argsort(-fitness(x))], '%10.5g') # save sort by fitness if bucket: os.system('gsutil cp evolve.txt gs://%s' % bucket) # upload evolve.txt