You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@mxnet.apache.org by zh...@apache.org on 2020/09/21 19:50:55 UTC

[incubator-mxnet] branch v1.x updated: Tweeking syntax to be closer to other tests (#19186)

This is an automated email from the ASF dual-hosted git repository.

zhasheng pushed a commit to branch v1.x
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git


The following commit(s) were added to refs/heads/v1.x by this push:
     new a76cc20  Tweeking syntax to be closer to other tests (#19186)
a76cc20 is described below

commit a76cc20e1244aa61d0d0a5e6b8a0a98e6e9ca03c
Author: Dick Carter <dc...@nvidia.com>
AuthorDate: Mon Sep 21 12:49:59 2020 -0700

    Tweeking syntax to be closer to other tests (#19186)
---
 R-package/tests/testthat/test_img_seg.R | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/R-package/tests/testthat/test_img_seg.R b/R-package/tests/testthat/test_img_seg.R
index 4af7d62..20d09f6 100644
--- a/R-package/tests/testthat/test_img_seg.R
+++ b/R-package/tests/testthat/test_img_seg.R
@@ -154,12 +154,11 @@ test_that("UNET", {
   train.y.array <- train.y
   dim(train.y.array) <- c(IMG_SIZE, IMG_SIZE, 1, 30)
   
-  devices <- mx.ctx.default()
   mx.set.seed(0)
   
   net <- get_unet()
   
   model <- mx.model.FeedForward.create(net, X = train.array, y = train.y.array, 
-    ctx = devices, num.round = 2, initializer = mx.init.normal(sqrt(2/576)), 
+    ctx = mx.ctx.default(), num.round = 2, initializer = mx.init.normal(sqrt(2/576)),
     learning.rate = 0.05, momentum = 0.99, array.batch.size = 2)
 })