Skip to content

Commit 5f2e03b

Browse files
Update README.md
1 parent 4a9ba5e commit 5f2e03b

File tree

1 file changed

+15
-16
lines changed

1 file changed

+15
-16
lines changed

README.md

Lines changed: 15 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -8,23 +8,22 @@ make build
88
```
99

1010
## Running the codes in the container
11-
1. Run a container and enter it.
11+
### Network training
12+
1. Example : Network training
1213
```
13-
make run
14+
docker-compose up pytorch-train
15+
```
16+
2. Visualization using Tensorboard
17+
```
18+
docker-compose up tensorboard
1419
```
1520

16-
2. Run a script inside the container
17-
1. Example : Training with greenhouse
18-
```
19-
cd /root/PyTorch-ENet
20-
python train.py -m train --save-dir save/dir --name [model_name] --dataset greenhouse --dataset-dir dataset/dir --with-unlabeled
21-
```
22-
1. Run tensorboard
23-
```
24-
cd /root/PyTorch-ENet
25-
tensorboard --logdir=./runs/
26-
```
27-
You can now use `docker-compose` to launch both the training script and `tensorboard`.
28-
```
29-
docker-compose up
21+
### ROS node
22+
1. Build (`catkin build`)
23+
```
24+
docker-compose up build
25+
```
26+
2. Run a ROS node (`roslaunch pytorch_enet_ros pytorch_enet_ros.launch`)
27+
```
28+
docker-compose up pytorch-ros-node
3029
```

0 commit comments

Comments
 (0)