diff --git a/README.md b/README.md index c6ea96f..e5653f6 100644 --- a/README.md +++ b/README.md @@ -23,7 +23,9 @@ These modules can be accessed via the ipython example notebooks or via the model > python DeepEnsemble.py low 10 /Users/rnevin/Documents/DeepUQ/ --save_final_checkpoints=True --savefig=True --n_epochs=10 -This command will train a 10 network, 10 epoch ensemble on the low noise data and will save figures and final checkpoints to the specified directory. For more information on the arguments: +This command will train a 10 network, 10 epoch ensemble on the low noise data and will save figures and final checkpoints to the specified directory. Required arguments are the noise setting (low/medium/high), the number of ensembles, and the working directory. + +For more information on the arguments: > python DeepEnsemble.py --help ## Installation