File tree 2 files changed +9
-3
lines changed
2 files changed +9
-3
lines changed Original file line number Diff line number Diff line change @@ -9,15 +9,15 @@ the Google NeuralHydrology package.
9
9
To build locally:
10
10
11
11
` 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} . `
13
13
14
14
15
15
## neuralhydrology/demo1
16
16
17
17
To build locally:
18
18
19
19
` 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} . `
21
21
22
22
Local demo for single basin train & test
23
23
@@ -60,3 +60,7 @@ docker run --rm -it --gpus all \
60
60
Then:
61
61
* ` python local_main.py -s train -d /data -e /experiments -b 02430085 `
62
62
* ` 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> `
Original file line number Diff line number Diff line change 1
1
"""
2
2
This script uses the demo1 container image to train and test neuralhyrology models.
3
+
4
+ On turtleland4, use venv ~/.py3-venv/neuralhydrology
3
5
"""
4
6
5
7
import argparse
12
14
# Container image constants
13
15
#[registry-hostname]/[username/organization-name]/[image-name]:[tag]
14
16
IMAGE_REGISTRY_USER = 'ghcr.io/johnkit'
15
- IMAGE_NAME = 'nh /demo1:latest '
17
+ IMAGE_NAME = 'neuralhydrology /demo1:1.0.0 '
16
18
17
19
# Optional filename for command line args
18
20
ARGS_FILENAME = '.args.txt'
You can’t perform that action at this time.
0 commit comments