visual-qa icon indicating copy to clipboard operation
visual-qa copied to clipboard

Error with Progressbar

Open aniemerg opened this issue 8 years ago • 0 comments

when running trainMLP.py, the training crashes in Keras' Progbar. The problem appears to be the following line 108: progbar.add(args.batch_size, values=[("train loss", loss)])

loss should probably be a float value of some sort, but when I look in the debugger, it is a list: [array(7.346083164215088, dtype=float32)]

The following change on line 108 appears to fix the issue: progbar.add(args.batch_size, values=[("train loss", loss[0].item(0))])

aniemerg avatar Mar 09 '16 02:03 aniemerg