Maybe dimension mistake

[PanPanZou]

Thanks for your nice implementation. But I meet this mistake when I sh train-omniglot-20way-1shot.sh ,is there any dimension dismatch? I don't know how to fix it. It's much appreciated if anyone can help me.
Traceback (most recent call last):
File "maml.py", line 230, in
main()
File "/usr/local/lib/python3.6/dist-packages/click/core.py", line 829, in call
return self.main(*args, **kwargs)
File "/usr/local/lib/python3.6/dist-packages/click/core.py", line 782, in main
rv = self.invoke(ctx)
File "/usr/local/lib/python3.6/dist-packages/click/core.py", line 1066, in invoke
return ctx.invoke(self.callback, **ctx.params)
File "/usr/local/lib/python3.6/dist-packages/click/core.py", line 610, in invoke
return callback(*args, **kwargs)
File "maml.py", line 227, in main
learner.train(exp)
File "maml.py", line 151, in train
mt_loss, mt_acc, mv_loss, mv_acc = self.test()
File "maml.py", line 113, in test
tloss, tacc = evaluate(test_net, train_loader)
File "/home/src-plot/score.py", line 30, in evaluate
loss += l.data.cpu().numpy()[0]
IndexError: too many indices for array