alexbw · July 17, 2018 16:34
diff --git a/autograph_huber_loss.py b/autograph_huber_loss.py
 @autograph.convert()
 def huber_loss(a):
  if tf.abs(a) <= delta:
    loss = a * a / 2
  else:
    loss = delta * (tf.abs(a) - delta / 2)
  return loss
	@autograph.convert()
	def huber_loss(a):
	if tf.abs(a) <= delta:
	loss = a * a / 2
	else:
	loss = delta * (tf.abs(a) - delta / 2)
	return loss