Update README.md

7d060e6e · mashun1 · 0b8209b3 · 7d060e6e
Commit 7d060e6e authored Jan 31, 2024 by mashun1
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

README.md README.md +4 -1

No files found.
--- a/README.md
+++ b/README.md
@@ -84,11 +84,14 @@

 注意：原始数据为`1024x1024`需要处理为`512x512`，可运行`data_process.py`对数据进行处理。

+本项目也提供了mini数据集（链接：https://pan.baidu.com/s/1HmaiwJygJMg7O-F9VaXURQ 
+提取码：kwai），可用于验证程序是否可以正常运行。
+
 ## 训练

 ### 阶段一 - 训练VQGAN

-    python -m torch.distributed.launch --nproc_per_node=8 --master_port=4321 basicsr/train.py -opt options/VQGAN_512_ds32_nearest_stage1.yml --launcher pytorch
+    python -m torch.distributed.launch --nproc_per_node=8 --master_port=4321 basicsr/train.py -opt options/VQGAN_512_ds32_nearest_stage1.yml        --launcher pytorch

 获取训练数据的码本序列，可以加速后续训练