Open mrsempress opened 4 years ago
I don't know in test.py lines 25: labels = make_variable(labels).squeeze_(), why the lables need squeeze_(), and in 'pretrain.py' the labels don't need squeeze(). Maybe because the USPS dataset?
test.py lines 25: labels = make_variable(labels).squeeze_()
I don't know in
test.py lines 25: labels = make_variable(labels).squeeze_()
, why the lables need squeeze_(), and in 'pretrain.py' the labels don't need squeeze(). Maybe because the USPS dataset?