optimization loss
Yang-L1 opened this issue · 1 comments
Yang-L1 commented
Hi,
Line 119 in train.py
train_op = optimizer.minimize(loss, global_step=batch)
Should this loss be loss[0]? Since the loss is a tuple object from the get_loss() function.
Btw, was the pretrained model trained by a huber loss or L2 loss?
xingyul commented
loss
is a scalar tensor. The pretrained model is trained under L2 loss.