UFDClassifierConfig

class UFDClassifierConfig(embed_dim=1024, num_class=2, initrange=0.1, **kwargs)

This is the configuration class to store the configuration of a UFDClassifierModel. It is used to instantiate a UFD Classifier model according to the specified arguments, defining the model architecture.

Parameters

PretrainedConfig (PretrainedConfig) – transformer PreTrainedConfig base class