
The test requires only 12GB GPU memory, while the training needs at least 80GB GPU memory. When the batch_size is less than 4 during training, it will seriously affect the convergence effect. We train 100 epochs by default. 
### Preparation

Download [ScanNet](http://www.scan-net.org/) v2.

Only files suffixed _vh_clean_2.ply and _vh_clean_2.labels.ply are needed.

Set the dataset path in line 9 of config.py. 

Set processed dataset path in line 11-13 if you want.

Then run prepare_scannetv2.py to process raw data into tensors.

```bash
python prepare_scannetv2.py
```
