--- title: Miscellaneous keywords: fastai sidebar: home_sidebar summary: "This contains a set of experiments." description: "This contains a set of experiments." nb_path: "nbs/140_models.misc.ipynb" ---
from tsai.models.TST import *
xb = torch.randn(16, 1, 1000)
model = InputWrapper(TST, 1, 4, 1000, 10, 224)
test_eq(model.to(xb.device)(xb).shape, (16,4))
from tsai.models.TST import *
xb = torch.randn(16, 1, 20)
model = RecursiveWrapper(TST(1, 1, 20), 5)
test_eq(model.to(xb.device)(xb).shape, (16, 5))