README.md
setup.py
datasets/__init__.py
datasets/cifar10/__init__.py
datasets/cifar10/data_loader.py
datasets/cifar10/main.py
datasets/cifar10/model_loader.py
datasets/cifar10/models/__init__.py
datasets/cifar10/models/densenet.py
datasets/cifar10/models/resnet.py
datasets/cifar10/models/vgg.py
datasets/cinic10/__init__.py
datasets/cinic10/data_loader.py
datasets/cinic10/main.py
datasets/cinic10/model_loader.py
datasets/cinic10/models/__init__.py
datasets/cinic10/models/baselines.py
loss_landscape/__init__.py
loss_landscape/data_loader.py
loss_landscape/evaluation.py
loss_landscape/h52vtp.py
loss_landscape/h5_util.py
loss_landscape/hess_vec_prod.py
loss_landscape/model_loader.py
loss_landscape/mpi4pytorch.py
loss_landscape/net_plotter.py
loss_landscape/plot_1D.py
loss_landscape/plot_2D.py
loss_landscape/plot_hessian_eigen.py
loss_landscape/plot_surface.py
loss_landscape/plot_trajectory.py
loss_landscape/projection.py
loss_landscape/scheduler.py
loss_landscape/utils.py
loss_landscape.egg-info/PKG-INFO
loss_landscape.egg-info/SOURCES.txt
loss_landscape.egg-info/dependency_links.txt
loss_landscape.egg-info/requires.txt
loss_landscape.egg-info/top_level.txt