--- 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" ---
{% raw %}
{% endraw %} {% raw %}
{% endraw %} {% raw %}

class InputWrapper[source]

InputWrapper(arch, c_in, c_out, seq_len, new_c_in=None, new_seq_len=None, **kwargs) :: Module

Same as nn.Module, but no need for subclasses to call super().__init__

{% endraw %} {% raw %}
{% endraw %} {% raw %}
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))
{% endraw %} {% raw %}

class ResidualWrapper[source]

ResidualWrapper(model) :: Module

Same as nn.Module, but no need for subclasses to call super().__init__

{% endraw %} {% raw %}
{% endraw %} {% raw %}

class RecursiveWrapper[source]

RecursiveWrapper(model, n_steps, anchored=False) :: Module

Same as nn.Module, but no need for subclasses to call super().__init__

{% endraw %} {% raw %}
{% endraw %} {% raw %}
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))
{% endraw %}