diff --git a/README.md b/README.md index 6aea256..dd92163 100644 --- a/README.md +++ b/README.md @@ -63,6 +63,11 @@ python3 src/train_edit.py --config_file configs/bedroom_sg_diffusion.yaml --outp You can change `config_file` to other room types. All config files are under [configs](./configs/) folder. +Pretained checkpoints download links for bedroom scene (batch_size used: 256): + +- SG: +- SG2SC: + ### 4. Evaluation To run the evaluation, we need both `SG_WEIGHT` and `SG2SC_WEIGHT`. Those weights should be found under `output_directory` for training.