modelzoo.transformers.pytorch.gpt2.utils.set_attention_kernel

modelzoo.transformers.pytorch.gpt2.utils.set_attention_kernel(params)

Set attention kernel related params :param params: model_params :return:

previous

modelzoo.transformers.pytorch.gpt2.utils

next

modelzoo.transformers.pytorch.gpt2.utils.set_defaults