CIPS-官方Pytorch实施
的纸张
要求
点安装-r requirements.txt
用法
首先创建lmdb数据集:
python prepare_data.py images --out LMDB_PATH --n_worker N_WORKER --size SIZE1,SIZE2,SIZE3,... DATASET_PATH
这会将图像转换为jpeg并预先调整其大小。
要在FFHQ-256或教堂上进行训练,请运行:
python3 -m torch.distributed.launch --nproc_per_node = 8 --master_port = 1234 train.py --n_sample = 8 --batch = 4 --fid_batch = 8 --Generator = CIPSskip --output_dir = skip- [ffhq
评论0
最新资源