tabensemb.model.TorchModelWrapper ================================= .. currentmodule:: tabensemb.model .. autoclass:: TorchModelWrapper :show-inheritance: .. rubric:: Methods .. automethod:: __init__ .. autosummary:: ~TorchModelWrapper.reset_forward ~TorchModelWrapper.wrap_forward .. HACK -- the point here is that we don't want this to appear in the output, but the autosummary should still generate the pages. .. autosummary:: :toctree: TorchModelWrapper.__call__ TorchModelWrapper.eval TorchModelWrapper.reset_forward TorchModelWrapper.wrap_forward .. HACK -- the point here is that we don't want this to appear in the output, but the autosummary should still generate the pages. .. autosummary:: :toctree: TorchModelWrapper.hidden_rep_dim TorchModelWrapper.hidden_representation