xuanzhu07's picture
Update README.md
2f324cb verified
|
Raw
History Blame Contribute Delete
950 Bytes
---
license: mit
---
# Deformable Convolution with Stochasticity - Model Weights
This repository contains pretrained model weights for the project [Deformable_Convolution_with_Stochasticity](https://github.com/theSleepyPig/Deformable_Convolution_with_Stochasticity).
## Available DCS Weights
| Dataset | Model | Filename |
|-----------|----------------|------------------------|
| CIFAR-10 | ResNet18 | `cifar10_resnet18.pth` |
| CIFAR-10 | WideResNet34 | `cifar10_wrn34.pth` |
| CIFAR-100 | ResNet18 | `cifar100_resnet18.pth`|
| CIFAR-100 | WideResNet34 | `cifar100_wrn34.pth` |
| CIFAR-10 | ViT-t+Conv | `vit_stage_one.pth` |
| CIFAR-10 | ViT-t+Conv | `vit_base.pth` |
| CIFAR-10 | ViT-t+DCS | `vit_dcs.pth` |
| ImageNet | ResNet50 | `r50_dcs.pth` |
More weights will be organized and released soon.
<!-- code in lab3, weight in lab 3 -->