Method __init__

SwissArmyTransformer/sat/model/official/dpr_model.py:81–84

(self, args, transformer=None, **kwargs)

class DPRReader(BaseModel):
    def __init__(self, args, transformer=None, **kwargs):
        super(DPRReader, self).__init__(args, transformer=transformer, **kwargs)
        self.add_mixin("dpr-type", DPRTypeMixin(args.num_types, args.hidden_size))
        self.add_mixin("dpr-final", DPRReaderFinalMixin(args.hidden_size, args.projection_dim))
        
    @classmethod
    def add_model_specific_args(cls, parser):