Class: Informers::GPT2LMHeadModel
- Inherits:
-
GPT2PreTrainedModel
- Object
- PreTrainedModel
- GPT2PreTrainedModel
- Informers::GPT2LMHeadModel
- Defined in:
- lib/informers/models.rb
Constant Summary
Constants inherited from PreTrainedModel
PreTrainedModel::MAIN_INPUT_NAME
Instance Attribute Summary
Attributes inherited from GPT2PreTrainedModel
#dim_kv, #num_heads, #num_layers
Attributes inherited from PreTrainedModel
Method Summary
Methods inherited from GPT2PreTrainedModel
Methods inherited from PreTrainedModel
#call, construct_session, from_pretrained, #generate, #initialize
Constructor Details
This class inherits a constructor from Informers::GPT2PreTrainedModel