HF_BaseModelWrapper {fastai}

HF_BaseModelWrapper

Description

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

Usage

HF_BaseModelWrapper(
  hf_model,
  output_hidden_states = FALSE,
  output_attentions = FALSE,
  ...
)

Arguments

hf_model

model

output_hidden_states

output hidden states

output_attentions

output attentions

...

additional arguments to pass

Value

None