From b3a8546e2c1472d0e669bd169cef091c2530c792 Mon Sep 17 00:00:00 2001 From: zhangmeng <775834166@qq.com> Date: 星期五, 13 十二月 2019 17:17:02 +0800 Subject: [PATCH] udpate --- work/sdk/humantrack.go | 8 +++++++- 1 files changed, 7 insertions(+), 1 deletions(-) diff --git a/work/sdk/humantrack.go b/work/sdk/humantrack.go index 7227f2c..7236e32 100644 --- a/work/sdk/humantrack.go +++ b/work/sdk/humantrack.go @@ -42,8 +42,14 @@ if gpu == -1 { gpu = gogpu.ValidGPU(2048) } + logo.Infof("start gpu %d, batch %d, flag %d\n", gpu, t.batchSize, t.flag) t.tracker = gohumantrack.NewHumanTracker(gpu, t.batchSize, t.flag) - return true + if t.tracker != nil { + logo.Infoln("Start Success") + return true + } + logo.Infoln("Start Failed") + return false } // Run impl -- Gitblit v1.8.0