We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
2 parents 42a9e7f + 60a4c83 commit 1a92a9fCopy full SHA for 1a92a9f
containers/BasicTerm_ME_python/main.py
@@ -3,7 +3,7 @@
3
def main():
4
parser = argparse.ArgumentParser(description="Term ME model runner")
5
parser.add_argument("--multiplier", type=int, default=100, help="Multiplier for model points")
6
- # add an argument that must be either "torch" or "jax"
+ # add an argument that must be either "torch_recursive" or "jax_iterative"
7
parser.add_argument("--model", type=str, default="jax_iterative", choices=["torch_recursive", "jax_iterative"], help="Model to run")
8
args = parser.parse_args()
9
0 commit comments