import tensorflow as tf


					return {
						"train":{
										"loss":loss, 
										"logits":logits,
										"train_op":train_op,
										"cross_entropy":label_loss,
										"distillation_loss":distillation_loss["distillation_loss"],
										"kd_num":tf.reduce_sum(features["distillation_ratio"]),
										"ce_num":tf.reduce_sum(features["label_ratio"]),
										"teacher_logit":teacher_logit,
										"student_logit":student_logit,
										"label_ratio":features["label_ratio"],
										"distilaltion_logits_loss":distillation_loss["distillation_logits_loss"],
										"distilaltion_feature_loss":distillation_loss["distillation_feature_loss"],
										"distillation_loss":distillation_loss["distillation_loss"],
										"st_accuracy":st_accuracy,
										"te_accuracy":te_accuracy,
