Tweeking syntax to be closer to other tests (apache#19186)

DickJC123 · Sep 21, 2020 · a98fd09 · a98fd09
1 parent a35d568
commit a98fd09
Showing 1 changed file with 1 addition and 2 deletions.
diff --git 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)
 })