Warmup learning rate wrapper for Pytorch Scheduler
-
Updated
Apr 12, 2020 - Python
Warmup learning rate wrapper for Pytorch Scheduler
Comprehensive image classification for training multilayer perceptron (MLP), LeNet, LeNet5, conv2, conv4, conv6, VGG11, VGG13, VGG16, VGG19 with batch normalization, ResNet18, ResNet34, ResNet50, MobilNetV2 on MNIST, CIFAR10, CIFAR100, and ImageNet1K.
Learning-rate schedules from boundary-value constrained trust-region problems.
Provide a collection of flexible learning rate schedulers for PyTorch with presets, warmup wrappers, and research-based implementations.
To associate your repository with the learning-rate-warmup topic, visit your repo's landing page and select "manage topics."