Final project for DD2412 Deep Learning, Advanced Course at KTH Royal Institute of Technology, Stockholm.
- Diogo Pinheiro (https://github.com/DiogorPinheiro)
- Jakob Lindén (https://github.com/jakobGTO)
- Tensorflow
- Numpy
- Robustness-metrics (source: https://github.com/google-research/robustness_metrics)
- Matplotlib
- Uncertainty Baselines (source: https://github.com/google/uncertainty-baselines)
-
Clone repository from https://github.com/google/uncertainty-baselines.git
-
Place script "run-tests" in the root directory of the cloned repository
-
Run command "chmod u+x run-tests" in that same directory
-
Run tests in root directory using
./run-tests --dataset cifar100 --model batchensemble --gpu True --download True --cores 4 --requirements False --epochs 10
The following parameters are currently provided: - dataset : "cifar100" or "cifar10"
- model : "all" (will run all pre-defined models) or write a specific one (e.g. "batchensemble" or "dropout") - gpu : Boolean that represents whether the process will be ran using gpu or not - download: If set to True it will download the required dataset to a temporary directory - cores : Number of cores to be user - requirements: If set to True it will first and foremost install all necessary dependencies - epochs : Number of epochs to be performed