↓ 2 callersFunctiontrain_batch(model, optimizer, scheduler,criterion,sp_loss_fct, batch, global_step)
CAIL2020/ydljtpu/train.py:174
↓ 2 callersFunctionvalid(model, dataset, epoch, writer, config, gpu_list, output_function, mode="valid")
CAIL2020/sfksz/tools/eval_tool.py:48
↓ 2 callersFunctionvalid(model, dataset, epoch, writer, config, gpu_list, output_function, mode="valid")
CAIL2020/sfkst/eval_tool.py:52
↓ 2 callersFunctionvalid(model, dataset, epoch, writer, config, gpu_list, output_function, mode="valid")
CAIL2020/sfksu/eval_tool.py:52
↓ 1 callersMethod__init__(
self,
initial_learning_rate,
decay_schedule_fn,
warmup_steps,
power=1.0,
CAIL2020/ydlj/baseline/transformers/optimization_tf.py:29
↓ 1 callersMethod_attn(self, q, k, v, attention_mask=None, head_mask=None)
CAIL2020/ydlj/baseline/transformers/modeling_gpt2.py:144
↓ 1 callersMethod_attn(self, q, k, v, attention_mask=None, head_mask=None)
CAIL2020/ydlj/baseline/transformers/modeling_openai.py:168