Skip to content

Commit

Permalink
Merge branch 'master' of https://github.com/WangYueFt/dgcnn
Browse files Browse the repository at this point in the history
  • Loading branch information
syb7573330 committed Feb 4, 2018
2 parents 1b19104 + c86da11 commit ddef0fb
Showing 1 changed file with 39 additions and 0 deletions.
39 changes: 39 additions & 0 deletions part_seg/Readme.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,39 @@
## Run part segmentation experiments

### Requirements

2 GPUs (each with 12 GB memory) for distributed training.

### Data preparation

Load the data for part segmentation.

```
sh +x download_data.sh
```

### Training

Train the model using 2 GPUs.

```
python train_multi_gpu.py
```

Model parameters are saved every 10 epochs in "train_results/trained_models/".

### Testing

To test the model parameter set saved after epoch n,

```
python test.py --model_path train_results/trained_models/epoch_n.ckpt
```

n = 0, 10, 20, ..., 190, 200.

For example, if we want to test the model saved after 160 epochs,

```
python test.py --model_path train_results/trained_models/epoch_160.ckpt
```

0 comments on commit ddef0fb

Please sign in to comment.