cache CI depednencies

.github/workflows/workflow.yml

@@ -31,16 +31,23 @@ jobs:
         python-version: [3.6]
         TF-version: [tensorflow==1.5.0, tensorflow-cpu==1.15.0]
     steps:
-      - uses: actions/checkout@v1
+      - uses: actions/checkout@v2
       - name: Set up Python ${{ matrix.python-version }}
-        uses: actions/setup-python@v1
+        uses: actions/setup-python@v2
         with:
           python-version: ${{ matrix.python-version }}
+      - name: Cache dependencies
+        uses: actions/cache@v2
+        with:
+          path: |
+            ${{ env.pythonLocation }}/lib/python3.6/site-packages
+          key: ${{ runner.os }}-${{ matrix.TF-version }}-20200711
+
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip
           python -m pip install scikit-image opencv-python lmdb h5py pyarrow
-          python -m pip install .
+          python -m pip install -e .
           python -c "import tensorpack.dataflow"
           # check that dataflow can be imported alone without tensorflow
           python -c "import cv2; print('OpenCV '+ cv2.__version__)"

examples/FasterRCNN/train.py

@@ -84,7 +84,7 @@ if __name__ == '__main__':
         HostMemoryTracker(),
         ThroughputTracker(samples_per_step=cfg.TRAIN.NUM_GPUS),
         EstimatedTimeLeft(median=True),
-        SessionRunTimeout(60000),   # 1 minute timeout
+        SessionRunTimeout(600000),   # 10 minute timeout
         GPUUtilizationTracker()
     ]
     if cfg.TRAIN.EVAL_PERIOD > 0:

sotabench/sotabench_setup.sh

@@ -5,8 +5,8 @@ set -v
 . /workspace/venv/bin/activate
 
 pip install -e .
-pip install tensorflow-gpu==1.14.0
-pip install opencv-python scipy
+pip install tensorflow-gpu==1.15.0
+pip install opencv-python-headless scipy pycocotools>=2.0.1
 
 echo "Extracting ..."
 cd ./.data/vision/coco