search

waldeland / CNN-for-ASI

Tutorial: Convolutional Neural Networks for Automated Seismic Interpretation
Other
149 stars 79 forks source link

Logger cannot deal with np.array #12

Open GGDRriedel opened 5 years ago

GGDRriedel commented 5 years ago

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)