Traceback (most recent call last): File "main.py", line 152, in train(epoch)<

IndexError: invalid index of a 0-dim tensor. Use `tensor.item()` in Python or `tensor.item<T>()` in C++ to convert a 0-dim tensor to a number about pytorch-slimming HOT 2 OPEN

foolwood commented on July 20, 2024

IndexError: invalid index of a 0-dim tensor. Use `tensor.item()` in Python or `tensor.item()` in C++ to convert a 0-dim tensor to a number

from pytorch-slimming.

Comments (2)

shine0801 commented on July 20, 2024

I have the same problem. Has it been solved?

from pytorch-slimming.

1337Eddy commented on July 20, 2024

That's a version problem. Replace 100. * batch_idx / len(train_loader), loss.data[0])) in line 120 with 100. * batch_idx / len(train_loader), loss.data.item())) and test_loss += F.cross_entropy(output, target, size_average=False).data[0] in line 131 with test_loss += F.cross_entropy(output, target, size_average=False).data.item()

from pytorch-slimming.

Related Issues (16)

model size after finetuning HOT 10
这个代码需要torch哪个版本？ HOT 4
python main.py -refine pruned.pth.tar --epochs 40
稀疏训练的时候，应该指定模型吧？
剪枝之后的模型无法正常检出框 HOT 1
[59, 64, 'M', 128, 128, 'M', 256, 256, 256, 256, 'M', 470, 239, 59, 40, 'M', 0, 3, 54, 483]，有一层卷积数量是0，怎么解决呢 HOT 4
Question about loss
Is not l1 loss the smooth l1 mentioned in the paper? HOT 1
Pruned error HOT 7
m.weight.grad.data.add_(args.s*torch.sign(m.weight.data)) # L1 HOT 2
error:torch.load HOT 1
Model save problem
稀疏化训练问题
剪枝程序vggprune.py遇见的问题 HOT 2
IndexError: index 0 is out of bounds for dimension 0 with size 0' in Prune.py HOT 1

IndexError: invalid index of a 0-dim tensor. Use `tensor.item()` in Python or `tensor.item<T>()` in C++ to convert a 0-dim tensor to a number about pytorch-slimming HOT 2 OPEN

Comments (2)

Related Issues (16)

Recommend Projects

React

Vue.js

Typescript

TensorFlow

Django

Laravel

D3

Recommend Topics

javascript

web

server

Machine learning

Visualization

Game

Recommend Org

Facebook

Microsoft

Google

Alibaba

D3

Tencent