Skip to content

Commit

Permalink
addding info on args
Browse files Browse the repository at this point in the history
  • Loading branch information
beckynevin committed Apr 1, 2024
1 parent 11d5e62 commit 9ae98ce
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 9ae98ce

Please sign in to comment.