Open buttercutter opened 3 years ago
Problem solved using https://github.com/promach/pt.darts/
First apply .contiguous() before view. Eg:
correct_k = correct[:k].view(-1).float().sum(0) must be changed to correct_k = correct[:k].contiguous().view(-1).float().sum(0)
Suggested by https://github.com/cezannec/capsule_net_pytorch/issues/4
I ran
python search.py --name cifar10 --dataset cifar10 --batch_size 32
, but I have the following error at utils.py. Why ?