├── .gitignore ├── LICENSE ├── README.md ├── config.py ├── dataset ├── __init__.py ├── dali_imagenet.py ├── imagenet_folder.py └── meter.py ├── loss ├── __init__.py └── label_smooth.py ├── optim ├── __init__.py ├── lamb.py ├── lars.py ├── lr_scheduler │ ├── __init__.py │ ├── cosine.py │ ├── delayed.py │ ├── multistep.py │ ├── onecycle.py │ └── poly.py └── utils.py ├── requirements.txt └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NUS-HPC-AI-Lab/pytorch-lamb/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NUS-HPC-AI-Lab/pytorch-lamb/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NUS-HPC-AI-Lab/pytorch-lamb/HEAD/README.md -------------------------------------------------------------------------------- /config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NUS-HPC-AI-Lab/pytorch-lamb/HEAD/config.py -------------------------------------------------------------------------------- /dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NUS-HPC-AI-Lab/pytorch-lamb/HEAD/dataset/__init__.py -------------------------------------------------------------------------------- /dataset/dali_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NUS-HPC-AI-Lab/pytorch-lamb/HEAD/dataset/dali_imagenet.py -------------------------------------------------------------------------------- /dataset/imagenet_folder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NUS-HPC-AI-Lab/pytorch-lamb/HEAD/dataset/imagenet_folder.py -------------------------------------------------------------------------------- /dataset/meter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NUS-HPC-AI-Lab/pytorch-lamb/HEAD/dataset/meter.py -------------------------------------------------------------------------------- /loss/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NUS-HPC-AI-Lab/pytorch-lamb/HEAD/loss/__init__.py -------------------------------------------------------------------------------- /loss/label_smooth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NUS-HPC-AI-Lab/pytorch-lamb/HEAD/loss/label_smooth.py -------------------------------------------------------------------------------- /optim/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NUS-HPC-AI-Lab/pytorch-lamb/HEAD/optim/__init__.py -------------------------------------------------------------------------------- /optim/lamb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NUS-HPC-AI-Lab/pytorch-lamb/HEAD/optim/lamb.py -------------------------------------------------------------------------------- /optim/lars.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NUS-HPC-AI-Lab/pytorch-lamb/HEAD/optim/lars.py -------------------------------------------------------------------------------- /optim/lr_scheduler/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NUS-HPC-AI-Lab/pytorch-lamb/HEAD/optim/lr_scheduler/__init__.py -------------------------------------------------------------------------------- /optim/lr_scheduler/cosine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NUS-HPC-AI-Lab/pytorch-lamb/HEAD/optim/lr_scheduler/cosine.py -------------------------------------------------------------------------------- /optim/lr_scheduler/delayed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NUS-HPC-AI-Lab/pytorch-lamb/HEAD/optim/lr_scheduler/delayed.py -------------------------------------------------------------------------------- /optim/lr_scheduler/multistep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NUS-HPC-AI-Lab/pytorch-lamb/HEAD/optim/lr_scheduler/multistep.py -------------------------------------------------------------------------------- /optim/lr_scheduler/onecycle.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NUS-HPC-AI-Lab/pytorch-lamb/HEAD/optim/lr_scheduler/onecycle.py -------------------------------------------------------------------------------- /optim/lr_scheduler/poly.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NUS-HPC-AI-Lab/pytorch-lamb/HEAD/optim/lr_scheduler/poly.py -------------------------------------------------------------------------------- /optim/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NUS-HPC-AI-Lab/pytorch-lamb/HEAD/optim/utils.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NUS-HPC-AI-Lab/pytorch-lamb/HEAD/requirements.txt -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NUS-HPC-AI-Lab/pytorch-lamb/HEAD/train.py --------------------------------------------------------------------------------