更新pytorch1.7 之后运行程序,有
utils.make_grid()
报下面的错误。
进入utils.py文件,把74行改成,就好用了
for idx, _ in enumerate([jj for jj in tensor]):
参考 https://github.com/pytorch/vision/issues/3025
2023-12-22 21:56:45
更新pytorch1.7 之后运行程序,有
utils.make_grid()
报下面的错误。
进入utils.py文件,把74行改成,就好用了
for idx, _ in enumerate([jj for jj in tensor]):
参考 https://github.com/pytorch/vision/issues/3025
下一篇:连连看