def learning_rate_with_decay(lr, global_step, discount_step, discount_factor):
	"""
	Near-optimal step decay learning rate schedule as proposed by https://arxiv.org/abs/1904.12838.
	"""
	return lr * discount_factor if global_step % discount_step == 0 and global_step > 0 else lr