├── Poster_ICML2022_Generalization_of_D_SGD_v8.png ├── README.md ├── Slides_ICML2022_Generalization_of_D_SGD.pdf ├── config.py ├── dataset ├── __init__.py ├── cifar10.py ├── cifar100.py ├── distributed_dataset.py └── tiny_imagenet.py ├── gpu_work.py ├── gpu_work ├── __init__.py └── worker.py └── models ├── __init__.py ├── resnet.py └── vgg.py /Poster_ICML2022_Generalization_of_D_SGD_v8.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CausalLearning/Generalization-of-DSGD/HEAD/Poster_ICML2022_Generalization_of_D_SGD_v8.png -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CausalLearning/Generalization-of-DSGD/HEAD/README.md -------------------------------------------------------------------------------- /Slides_ICML2022_Generalization_of_D_SGD.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CausalLearning/Generalization-of-DSGD/HEAD/Slides_ICML2022_Generalization_of_D_SGD.pdf -------------------------------------------------------------------------------- /config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CausalLearning/Generalization-of-DSGD/HEAD/config.py -------------------------------------------------------------------------------- /dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CausalLearning/Generalization-of-DSGD/HEAD/dataset/__init__.py -------------------------------------------------------------------------------- /dataset/cifar10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CausalLearning/Generalization-of-DSGD/HEAD/dataset/cifar10.py -------------------------------------------------------------------------------- /dataset/cifar100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CausalLearning/Generalization-of-DSGD/HEAD/dataset/cifar100.py -------------------------------------------------------------------------------- /dataset/distributed_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CausalLearning/Generalization-of-DSGD/HEAD/dataset/distributed_dataset.py -------------------------------------------------------------------------------- /dataset/tiny_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CausalLearning/Generalization-of-DSGD/HEAD/dataset/tiny_imagenet.py -------------------------------------------------------------------------------- /gpu_work.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CausalLearning/Generalization-of-DSGD/HEAD/gpu_work.py -------------------------------------------------------------------------------- /gpu_work/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CausalLearning/Generalization-of-DSGD/HEAD/gpu_work/__init__.py -------------------------------------------------------------------------------- /gpu_work/worker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CausalLearning/Generalization-of-DSGD/HEAD/gpu_work/worker.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CausalLearning/Generalization-of-DSGD/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CausalLearning/Generalization-of-DSGD/HEAD/models/resnet.py -------------------------------------------------------------------------------- /models/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CausalLearning/Generalization-of-DSGD/HEAD/models/vgg.py --------------------------------------------------------------------------------