├── .gitignore ├── LICENSE ├── README.md ├── benchmark.py ├── figures ├── scheme_compare_gradient_checkpoint.png └── table_compare_gradient_checkpoint.png ├── graph.py ├── manual_parse_graph.ipynb ├── net ├── alexnet.py ├── darts │ ├── genotypes.py │ ├── model.py │ ├── operations.py │ └── utils.py ├── densenet.py ├── inception.py ├── layer.py ├── model_factory.py ├── resnet.py └── vgg.py ├── requirements.txt ├── solver.py ├── train_cifar10.ipynb └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianweif/OptimalGradCheckpointing/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianweif/OptimalGradCheckpointing/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianweif/OptimalGradCheckpointing/HEAD/README.md -------------------------------------------------------------------------------- /benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianweif/OptimalGradCheckpointing/HEAD/benchmark.py -------------------------------------------------------------------------------- /figures/scheme_compare_gradient_checkpoint.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianweif/OptimalGradCheckpointing/HEAD/figures/scheme_compare_gradient_checkpoint.png -------------------------------------------------------------------------------- /figures/table_compare_gradient_checkpoint.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianweif/OptimalGradCheckpointing/HEAD/figures/table_compare_gradient_checkpoint.png -------------------------------------------------------------------------------- /graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianweif/OptimalGradCheckpointing/HEAD/graph.py -------------------------------------------------------------------------------- /manual_parse_graph.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianweif/OptimalGradCheckpointing/HEAD/manual_parse_graph.ipynb -------------------------------------------------------------------------------- /net/alexnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianweif/OptimalGradCheckpointing/HEAD/net/alexnet.py -------------------------------------------------------------------------------- /net/darts/genotypes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianweif/OptimalGradCheckpointing/HEAD/net/darts/genotypes.py -------------------------------------------------------------------------------- /net/darts/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianweif/OptimalGradCheckpointing/HEAD/net/darts/model.py -------------------------------------------------------------------------------- /net/darts/operations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianweif/OptimalGradCheckpointing/HEAD/net/darts/operations.py -------------------------------------------------------------------------------- /net/darts/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianweif/OptimalGradCheckpointing/HEAD/net/darts/utils.py -------------------------------------------------------------------------------- /net/densenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianweif/OptimalGradCheckpointing/HEAD/net/densenet.py -------------------------------------------------------------------------------- /net/inception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianweif/OptimalGradCheckpointing/HEAD/net/inception.py -------------------------------------------------------------------------------- /net/layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianweif/OptimalGradCheckpointing/HEAD/net/layer.py -------------------------------------------------------------------------------- /net/model_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianweif/OptimalGradCheckpointing/HEAD/net/model_factory.py -------------------------------------------------------------------------------- /net/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianweif/OptimalGradCheckpointing/HEAD/net/resnet.py -------------------------------------------------------------------------------- /net/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianweif/OptimalGradCheckpointing/HEAD/net/vgg.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianweif/OptimalGradCheckpointing/HEAD/requirements.txt -------------------------------------------------------------------------------- /solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianweif/OptimalGradCheckpointing/HEAD/solver.py -------------------------------------------------------------------------------- /train_cifar10.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianweif/OptimalGradCheckpointing/HEAD/train_cifar10.ipynb -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianweif/OptimalGradCheckpointing/HEAD/utils.py --------------------------------------------------------------------------------