pytorch-classification
pytorch-classification copied to clipboard
Classification with PyTorch.
Similar benchmarks on MNIST dataset would be great.
I found the layers of resnet-110 is 164, why called it resnet110 but not resnet164?
/pytorch-classification/utils/eval.py", line 16, in accuracy correct_k = correct[:k].view(-1).float().sum(0) RuntimeError: view size is not compatible with input tensor's size and stride (at least one dimension spans across two contiguous subspaces). Use...
This pull request includes several important updates and fixes to ensure compatibility with the latest version of PyTorch and to resolve issues related to tensor operations while training AlexNet on...