what's your version of pytorch? And could you provide train_label.txt and other three?

[LeviChenhh]

When I run the GDH.py,there is a bug that show "RuntimeError: size mismatch, m1: [8 x 512], m2: [2048 x 64] at C:/w/1/s/tmp_conda_3.6_041836/conda/conda-bld/pytorch_1556684464974/work/aten/src\THC/generic/THCTensorMathBlas.cu:268" . Some people say it's error of retract. But after my try, I ruled that out. Then I suspect it's a version problem.

Another question: In your code, there are four files: train_img.txt,train_label.txt,test_img.txt,test_label.txt could you provide the four files?

[LeviChenhh]

my e-mail is [email protected]