RuntimeError adding keepdim

mvasil / fashion-compatibility

Learning Type-Aware Embeddings for Fashion Compatibility

BSD 3-Clause "New" or "Revised" License

152 stars 42 forks source link

RuntimeError adding keepdim #13

Closed qa276390 closed 5 years ago

qa276390 commented 5 years ago

Since torchvision is not compatible with Pytorch version 0.1.12, I am running this project with torch==1.1.0. And the error occur in type_specific_network.py line 119 and 142. Adding keepdim=True to torch.normcould prevent from that. Also, in polyvore_outfits.py line 235 add list(...) might good.

RuntimeError: The expanded size of the tensor (64) must  match the existing  size (66) at non-singleton dimension 2. Target sizes: [256, 66, 64]. Tensor sizes: [256, 66]

fuhailin commented 5 years ago

Can you run this code without error? I want to consult you

qa276390 commented 5 years ago

@fuhailin finally, I install Pytorch==0.1.12 and torchvision==0.1.6 to run the codes successfully.