From fef7edf261f35f816465f8036777f9519117a7da Mon Sep 17 00:00:00 2001
From: Zhen Liu <liuzhen1994@hotmail.com>
Date: Wed, 2 Aug 2023 02:25:36 +0200
Subject: [PATCH] Update README.md

---
 README.md | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/README.md b/README.md
index 404cd75..b5657c0 100644
--- a/README.md
+++ b/README.md
@@ -143,6 +143,26 @@ python main_diffusion.py --mode=train --config=$DIFFUSION_CONFIG \
 
 where `$TRAIN_SPLIT_FILE` is a json list of indices to be included in the training set. Examples in `metadata/train_split/`. For the diffusion model config file, please refer to `configs/res64.py` or `configs/res128.py`.
 
+#### Training with our dataset
+
+The provided datasets are stored in `.npy` instead of `.pt`. Run the following instead
+
+```
+cd ../metadata/
+python save_meta.py --data_path $DMTET_NPY_FOLDER --json_path $META_FILE --extension npy
+```
+
+Train a diffusion model
+
+```
+cd ..
+
+python main_diffusion.py --mode=train --config=$DIFFUSION_CONFIG \
+--config.data.meta_path=$META_FILE \
+--config.data.filter_meta_path=$TRAIN_SPLIT_FILE \
+--config.data.extension=npy
+```
+
 ## Texture Generation
 
 Follow the instructions in https://github.com/TEXTurePaper/TEXTurePaper and create text-conditioned textures for the generated meshes.