HF_BaseModelWrapper: HF_BaseModelWrapper

In fastai: Interface to '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

fastai documentation built on March 31, 2023, 11:41 p.m.