HF_BaseModelWrapper: HF_BaseModelWrapper

Description Usage Arguments Value

View source: R/blurr_hugging_face.R

Description

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

Usage

1
2
3
4
5
6
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 July 28, 2021, 5:06 p.m.