From 9ff40baa4f0b005d50c4e3302b4103551160943f Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Wed, 07 Feb 2018 22:04:40 +0000
Subject: [PATCH] Fixes for training Yolo on small objects
---
.circleci/config.yml | 7 +++++--
1 files changed, 5 insertions(+), 2 deletions(-)
diff --git a/.circleci/config.yml b/.circleci/config.yml
index 9cf396b..4a79b0a 100644
--- a/.circleci/config.yml
+++ b/.circleci/config.yml
@@ -2,8 +2,11 @@
jobs:
build:
docker:
- - image: nvidia/cuda:9.0-cudnn7-devel-ubuntu16.04
+ - image: alantrrs/cuda-opencv:latest
+# - image: nvidia/cuda:9.0-cudnn7-devel-ubuntu16.04
working_directory: ~/work
steps:
- checkout
- - run: make LIBSO=1 GPU=1 CUDNN=1 -j 8
+ - run: make LIBSO=1 GPU=0 CUDNN=0 OPENCV=0 -j 8
+ - run: make clean
+ - run: make LIBSO=1 GPU=1 CUDNN=1 OPENCV=1 -j 8
--
Gitblit v1.10.0