Logger cannot deal with np.array

[GGDRriedel]

Line 98 in train.py needs to be changed from

logger.log_scalar('training_loss', var_to_np(loss),i)

to
logger.log_scalar('training_loss', var_to_np(loss).tolist(),i)