wohlert/generative-query-network-pytorch

error in notebook

wasif26 opened this issue · 1 comments

in ()
1 f, axarr = plt.subplots(1, 15, figsize=(20, 7))
2 for i, ax in enumerate(axarr.flat):
----> 3 ax.imshow(x[0][i].permute(1, 2, 0))

RuntimeError: number of dims don't match in permute

Yes, the notebooks haven't been updated to reflect the new codebase.