This code is adopted from one of the PyTorch Lightning examples and this PonderNet implementation.
I'm playing with PyTorch on the CIFAR10 dataset.
- Python 3.6+
- PyTorch 1.0+
# Start training INTERPOLATION TASK with:
python interpolation.py
# Start training EXTRAPOLATION TASK with:
python extrapolation.py
INTERPOLATION TASK
| Model | Acc. |
|---|---|
| ResNet18 | 93.02% |
| ResNet18 + PonderNet (epochs=10) | 83.24% |
| ResNet18 + PonderNet (epochs=50) | 90.96% |