Fix View conversion - Githubissues

clcarwin / convert_torch_to_pytorch

Convert torch t7 model to pytorch model and source.

MIT License

539 stars 161 forks source link

Fix View conversion #1

Closed CeLuigi closed 7 years ago

CeLuigi commented 7 years ago

The older version produced a flattened version of the input tensor including batch dimension. Specifically, given the 'pool5' output tensor of VGG16 net, i.e. batch_size x 512 x 7 x 7:

older version (Lambda(lambda x,s=torch.Size([-1]): x.view(s))): batch_size 512 7 * 7
new version (Lambda(lambda x: x.view(x.size(0),-1))): batch_size x 512 7 7

Thank you for your script

clcarwin commented 7 years ago

Thanks.