From dc513d389cae7749c2975292fe8e71b3f7c01285 Mon Sep 17 00:00:00 2001 From: YiYi Xu Date: Sun, 25 Sep 2022 20:09:06 -1000 Subject: [PATCH] Update README.md --- README.md | 15 ++------------- 1 file changed, 2 insertions(+), 13 deletions(-) diff --git a/README.md b/README.md index 97255fa..3c8ae61 100644 --- a/README.md +++ b/README.md @@ -123,23 +123,12 @@ python3 main.py --workdir=./fashion-mnist --mode=train --config=configs/fashion_ ``` W&B project page: [ddpm-flax-fashion-mnist](https://wandb.ai/yiyixu/ddpm-flax-fashion-mnist?workspace=user-yiyixu) -### oxford_flowers102 (on-going experiments) +### oxford_flowers102 -- run the model in default setting -``` -python3 main.py --workdir=./flower102--mode=train --config=configs/oxford102.py -``` - -- run the model with p2-weighting - -``` -python3 main.py --workdir=./flower102--mode=train --config=configs/oxford102.py --config.ddpm.p2_loss_weight_gamma=1. -``` -- with self-condition ``` -python3 main.py --workdir=./flower102--mode=train --config=configs/oxford102.py --config.ddpm.self_condition=True +python3 main.py --workdir=./flower102--mode=train --config=configs/oxford102_p2_selfcondition.py ``` W&B project page: [ddpm-flax-flower102](https://wandb.ai/yiyixu/ddpm-flax-flower102?workspace=user-yiyixu)