inst/tinytest/test-transformer-xl.R

model <- transformer(architecture = "Transformer-XL", model_name = "transfo-xl-wt103")
expect_true(inherits(model, "Transformer"))
expect_true(inherits(predict(model, "testing out", type = "embed-sentence", trace = FALSE), "matrix"))

## Clean up models folder
unlink_golgotha()
bnosac/golgotha documentation built on May 28, 2020, 4:06 a.m.