path # lines of code analyze_results/cifar/one_dimensional_subspaces.py 109 analyze_results/tinyimagenet/one_dimensional_subspaces.py 109 analyze_results/tinyimagenet/simplexes.py 55 args.py 231 data/__init__.py 4 data/cifar.py 59 data/imagenet.py 73 data/imagenetc.py 40 data/tinyimagenet.py 45 experiment_configs/cifar10/ensembles/eval_ensembles.py 55 experiment_configs/cifar10/ensembles/train_ensemble_members.py 26 experiment_configs/cifar10/one_dimensional_subspaces/eval_curves.py 77 experiment_configs/cifar10/one_dimensional_subspaces/eval_lines.py 77 experiment_configs/cifar10/one_dimensional_subspaces/eval_lines_layerwise.py 77 experiment_configs/cifar10/one_dimensional_subspaces/train_curves.py 41 experiment_configs/cifar10/one_dimensional_subspaces/train_lines.py 41 experiment_configs/cifar10/one_dimensional_subspaces/train_lines_layerwise.py 41 experiment_configs/cifar10/simplexes/train_simplexes.py 41 experiment_configs/cifar10/simplexes/train_simplexes_layerwise.py 41 experiment_configs/imagenet/ensembles/eval_ensembles.py 59 experiment_configs/imagenet/ensembles/train_ensemble_members.py 32 experiment_configs/imagenet/one_dimensional_subspaces/eval_lines.py 78 experiment_configs/imagenet/one_dimensional_subspaces/eval_lines_layerwise.py 78 experiment_configs/imagenet/one_dimensional_subspaces/train_lines.py 43 experiment_configs/imagenet/one_dimensional_subspaces/train_lines_layerwise.py 43 experiment_configs/tinyimagenet/ensembles/eval_ensembles.py 56 experiment_configs/tinyimagenet/ensembles/train_ensemble_members.py 29 experiment_configs/tinyimagenet/one_dimensional_subspaces/eval_curves.py 78 experiment_configs/tinyimagenet/one_dimensional_subspaces/eval_lines.py 79 experiment_configs/tinyimagenet/one_dimensional_subspaces/eval_lines_layerwise.py 78 experiment_configs/tinyimagenet/one_dimensional_subspaces/train_curves.py 42 experiment_configs/tinyimagenet/one_dimensional_subspaces/train_lines.py 42 experiment_configs/tinyimagenet/one_dimensional_subspaces/train_lines_layerwise.py 42 experiment_configs/tinyimagenet/simplexes/train_simplexes.py 42 experiment_configs/tinyimagenet/simplexes/train_simplexes_layerwise.py 42 main.py 194 models/__init__.py 14 models/builder.py 156 models/cifar_resnet.py 75 models/cifar_resnet_dropout.py 77 models/codegen.py 55 models/init.py 3 models/modules.py 117 models/modules_gen.py 521 models/resnet.py 163 models/tinyimagenetresnet.py 143 schedulers.py 191 trainers/__init__.py 10 trainers/average_weights.py 63 trainers/default.py 73 trainers/ensemble.py 98 trainers/eval_one_dim_subspaces.py 75 trainers/eval_one_dim_subspaces_multigpu.py 164 trainers/linestats_swa.py 108 trainers/random_average_weights_global.py 50 trainers/random_average_weights_layerwise.py 50 trainers/random_average_weights_perweight.py 70 trainers/simplex_ensembles.py 92 trainers/swa_endpoint_ensembles.py 68 trainers/swag.py 56 trainers/train_one_dim_subspaces.py 188 trainers/train_simplexes.py 165 utils.py 134 viz/config.py 44 viz/utils.py 381