cifkao Compute loss in predict.py