HF_BaseModelWrapper: HF_BaseModelWrapper

View source: R/blurr_hugging_face.R

HF_BaseModelWrapperR Documentation

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 21, 2022, 9:07 a.m.