update utils.create_pretrained()

Dockerfile

@@ -1,5 +1,6 @@
 # Start FROM Nvidia PyTorch image https://ngc.nvidia.com/catalog/containers/nvidia:pytorch
 FROM nvcr.io/nvidia/pytorch:20.03-py3
+RUN pip install -U gsutil
 
 # Create working directory
 RUN mkdir -p /usr/src/app
@@ -10,7 +11,6 @@ COPY . /usr/src/app
 
 # Install dependencies (pip or conda)
 #RUN pip install -r requirements.txt
-RUN pip install -U gsutil
 
 # Copy weights
 #RUN python3 -c "from models import *; \
@@ -41,7 +41,7 @@ RUN pip install -U gsutil
 
 # Bash into running container
 # sudo docker container exec -it ba65811811ab bash
-# python -c "from utils.utils import *; create_backbone('weights/last.pt')" && gsutil cp weights/backbone.pt gs://*
+# python -c "from utils.utils import *; create_pretrained('weights/last.pt')" && gsutil cp weights/pretrained.pt gs://*
 
 # Bash into stopped container
 # sudo docker commit 6d525e299258 user/test_image && sudo docker run -it --gpus all --ipc=host -v "$(pwd)"/coco:/usr/src/coco --entrypoint=sh user/test_image