search

sksq96 / pytorch-summary

Model summary in PyTorch similar to `model.summary()` in Keras
MIT License
4.01k stars 412 forks source link

TypeError: rand() argument after * must be an iterable, not int #139

Open Lin-plax opened 4 years ago

Lin-plax commented 4 years ago

net = models.resnet50() summary(net, input_size=([3, 224, 224]))

x = [torch.rand(2, in_size).type(dtype) for in_size in input_size] TypeError: rand() argument after must be an iterable, not int

Thanks for help!

TylerYep commented 4 years ago

See #124