tensorwatch
tensorwatch copied to clipboard
IndexError: too many indices for tensor of dimension 4
tw.draw_model(model,[1,3,300,300])
IndexError: too many indices for tensor of dimension 4