MCPcopy
hub / github.com/kuangliu/pytorch-cifar

github.com/kuangliu/pytorch-cifar @main sqlite

repository ↗ · DeepWiki ↗
220 symbols 383 edges 21 files 13 documented · 6%
README

Train CIFAR10 with PyTorch

I'm playing with PyTorch on the CIFAR10 dataset.

Prerequisites

  • Python 3.6+
  • PyTorch 1.0+

Training

# Start training with: 
python main.py

# You can manually resume the training with: 
python main.py --resume --lr=0.01

Accuracy

Model Acc.
VGG16 92.64%
ResNet18 93.02%
ResNet50 93.62%
ResNet101 93.75%
RegNetX_200MF 94.24%
RegNetY_400MF 94.29%
MobileNetV2 94.43%
ResNeXt29(32x4d) 94.73%
ResNeXt29(2x64d) 94.82%
SimpleDLA 94.89%
DenseNet121 95.04%
PreActResNet18 95.11%
DPN92 95.16%
DLA 95.47%

Core symbols most depended-on inside this repo

_make_layer
called by 4
models/senet.py
_make_layer
called by 4
models/dpn.py
_make_dense_layers
called by 4
models/densenet.py
swish
called by 4
models/efficientnet.py
_make_layer
called by 4
models/preact_resnet.py
_make_layer
called by 4
models/resnet.py
_make_layer
called by 4
models/regnet.py
_make_layer
called by 3
models/resnext.py

Shape

Method 115
Function 55
Class 50

Languages

Python100%

Modules by API surface

models/shufflenetv2.py17 symbols
models/pnasnet.py17 symbols
models/resnet.py16 symbols
models/preact_resnet.py16 symbols
models/densenet.py16 symbols
models/regnet.py14 symbols
models/efficientnet.py14 symbols
models/shufflenet.py13 symbols
models/dla_simple.py13 symbols
models/dla.py13 symbols
models/senet.py12 symbols
models/resnext.py12 symbols

For agents

$ claude mcp add pytorch-cifar \
  -- python -m otcore.mcp_server <graph>

⬇ download graph artifact