ernie

class Ernie(model_name, num_classes, task=None, **kwargs)

Bases: paddle.fluid.dygraph.layers.Layer

forward(input_ids, token_type_ids=None, position_ids=None, attention_mask=None)

Defines the computation performed at every call. Should be overridden by all subclasses.

Parameters

• *inputs (tuple) – unpacked tuple arguments

• **kwargs (dict) – unpacked dict arguments