TypeError: item() takes no arguments (1 given)

Eric-mingjie / network-slimming

Network Slimming (Pytorch) (ICCV 2017)

MIT License

907 stars 214 forks source link

Open Rhett-Lin opened 2 years ago

Rhett-Lin commented 2 years ago

运行main.py时报错

- batch_idx / len(train_loader), loss.item(0))) TypeError: item() takes no arguments (1 given)

Eric-mingjie commented 2 years ago

参考https://github.com/Eric-mingjie/network-slimming/issues/75