modelzoo.transformers.pytorch.gpt2.utils

Functions

set_attention_kernel	Set attention kernel related params :param params: model_params :return:
set_defaults	Update any missing parameters in the params dictionary with default values

previous

modelzoo.transformers.pytorch.gpt2.sparse_mask.create_fixed_sparse_attention_mask

next

modelzoo.transformers.pytorch.gpt2.utils.set_attention_kernel