QizhongYao issues

Repositories
Issues
Comments

Results 1 issues of


                                            QizhongYao

inception_v3 of vision 0.3.0 does not fit in DataParallel of torch 1.1.0

Environment: Python 3.5 torch 1.1.0 torchvision 0.3.0 Reproducible example: `import torch` `import torchvision` `model = torchvision.models.inception_v3().cuda()` `model = torch.nn.DataParallel(model, [0, 1])` `x = torch.rand((8, 3, 299, 299)).cuda()` `model.forward(x)` Error: >...

bug

needs discussion

module: models

topic: classification