From c2df97faacaa4fcee4c4ee3e79fe4e325023e78e Mon Sep 17 00:00:00 2001 From: Dick Carter Date: Tue, 22 Sep 2020 09:55:44 -0700 Subject: [PATCH] Tweeking syntax to be closer to other tests (#19186) (#19206) --- 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 4af7d62cf533..20d09f65f7c4 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) })