Skip to content

Commit 81cf13e

Browse files
committed
Update standalone demo script
1 parent 0d16dba commit 81cf13e

File tree

2 files changed

+9
-3
lines changed

2 files changed

+9
-3
lines changed

README.md

+6-2
Original file line numberDiff line numberDiff line change
@@ -9,15 +9,15 @@ the Google NeuralHydrology package.
99
To build locally:
1010

1111
`cd neuralhydrology`
12-
`VERSION=$(cat neuralhydrology.version.txt) && docker build -f neuralhydrology.dockerfile -t neuralhydrology:${VERSION} .`
12+
`VERSION=$(cat neuralhydrology.version.txt) && docker build -f neuralhydrology.dockerfile -t neuralhydrology/base:${VERSION} .`
1313

1414

1515
## neuralhydrology/demo1
1616

1717
To build locally:
1818

1919
`cd neuralhydrology/demo1`
20-
`VERSION=$(cat demo1.version.txt) && docker build -f demo1.dockerfile -t demo1:${VERSION} .`
20+
`VERSION=$(cat demo1.version.txt) && docker build -f demo1.dockerfile -t neuralhydrology/demo1:${VERSION} .`
2121

2222
Local demo for single basin train & test
2323

@@ -60,3 +60,7 @@ docker run --rm -it --gpus all \
6060
Then:
6161
* `python local_main.py -s train -d /data -e /experiments -b 02430085`
6262
* `python local_main.py -s test -d /data -e /experiments -b 02430085 -r <RUN_ID>`
63+
64+
To run the standalone demo:
65+
`cd neuralhydrology`
66+
`python demo.py -s train -d /data -e /experiments -b 02430085 -r <RUN_ID>`

neuralhydrology/demo1.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,7 @@
11
"""
22
This script uses the demo1 container image to train and test neuralhyrology models.
3+
4+
On turtleland4, use venv ~/.py3-venv/neuralhydrology
35
"""
46

57
import argparse
@@ -12,7 +14,7 @@
1214
# Container image constants
1315
#[registry-hostname]/[username/organization-name]/[image-name]:[tag]
1416
IMAGE_REGISTRY_USER = 'ghcr.io/johnkit'
15-
IMAGE_NAME = 'nh/demo1:latest'
17+
IMAGE_NAME = 'neuralhydrology/demo1:1.0.0'
1618

1719
# Optional filename for command line args
1820
ARGS_FILENAME = '.args.txt'

0 commit comments

Comments
 (0)