tests/testthat/test_PipeOpTorchConv.R

In mlr3torch: Deep Learning with 'mlr3'

test_that("PipeOpTorchConv1 autotest", {
  po_conv = po("nn_conv1d", kernel_size = 2, out_channels = 3)
  task = tsk("iris")
  graph = po("torch_ingress_num") %>>% po("nn_unsqueeze", dim = 2) %>>% po_conv

  expect_pipeop_torch(graph, "nn_conv1d", task)
})

test_that("PipeOpTorchConv1d paramtest", {
  res = expect_paramset(po("nn_conv1d"), nn_conv1d, exclude = "in_channels")
  expect_paramtest(res)
})

test_that("PipeOpTorchConv2 autotest", {
  po_conv = po("nn_conv2d", kernel_size = 2, out_channels = 2)
  task = nano_imagenet()
  graph = po("torch_ingress_ltnsr") %>>% po_conv

  expect_pipeop_torch(graph, "nn_conv2d", task)
})

test_that("PipeOpTorchConv2d paramtest", {
  res = expect_paramset(po("nn_conv2d"), nn_conv2d, exclude = "in_channels")
  expect_paramtest(res)
})

test_that("PipeOpTorchConv3 autotest", {
  po_conv = po("nn_conv3d", kernel_size = 2, out_channels = 2)
  task = nano_imagenet()
  graph = po("torch_ingress_ltnsr") %>>%
    po("nn_unsqueeze", dim = 5) %>>%
    po("nn_reshape", shape = c(-1, 3, 64, 8, 8)) %>>% po_conv

  expect_pipeop_torch(graph, "nn_conv3d", task)
})

test_that("PipeOpTorchConv3d paramtest", {
  res = expect_paramset(po("nn_conv3d"), nn_conv3d, exclude = "in_channels")
  expect_paramtest(res)
})


sampler_conv = function(dim, batch = TRUE) {
  list(
    conv_dim = dim,
    shape_in = sample(20:25, size = dim + 1 + as.integer(batch), replace = TRUE),
    out_channels = sample(1:3, size = 1, replace = TRUE),
    kernel_size = sample(5:6, size = dim, replace = TRUE),
    stride = sample(1:3, size = dim, replace = TRUE),
    padding = sample(1:2, size = dim, replace = TRUE),
    dilation = sample(1:2, size = dim, replace = TRUE),
    padding_mode = sample(c("zeros", "reflect", "replicate"), 1)
    # there is something wrong with circular padding: https://github.com/mlverse/torch/issues/940
  )
}

test_that("conv_output_shape works", {
  for (dim in 1:3) {
    testcase = sampler_conv(dim)
    mg = switch(dim,
      nn_conv1d,
      nn_conv2d,
      nn_conv3d,
    )
    args = testcase[names(testcase) %in% formalArgs(mg)]
    args$in_channels = testcase$shape_in[2L]
    m = do.call(mg, args = args)
    outshape = with_no_grad(m(do.call(torch_randn, args = list(unname(testcase$shape_in)))))$shape
    args1 = testcase[names(testcase) %in% formalArgs(conv_output_shape)]
    expect_true(all(outshape == do.call(conv_output_shape, args = args1)))
  }
})

test_that("conv_output_shape works", {
  for (dim in 1:3) {
    testcase = sampler_conv(dim, batch = FALSE)
    mg = switch(dim,
      nn_conv1d,
      nn_conv2d,
      nn_conv3d,
    )
    args = testcase[names(testcase) %in% formalArgs(mg)]
    args$in_channels = testcase$shape_in[1L]
    m = do.call(mg, args = args)
    outshape = with_no_grad(m(do.call(torch_randn, args = list(unname(testcase$shape_in)))))$shape
    args1 = testcase[names(testcase) %in% formalArgs(conv_output_shape)]
    expect_warning(shape <<- do.call(conv_output_shape, args = args1), regexp = "batch dimension")
    expect_true(all(outshape == shape))
  }
})