ModelZoo-PyTorch/PyTorch/built-in/mm/PIDM/utils · Ascend/ModelZoo-PyTorch - AtomGit

Ii-robot!6719 [built-in][Pytorch] 调整多模态模型存放目录

cf2aeb79创建于 2024年9月6日历史提交

文件	最后提交记录	最后更新时间
README.md	!6719 [built-in][Pytorch] 调整多模态模型存放目录 Merge pull request !6719 from zhangjunyi08/master	1 年前
gen.py	!6719 [built-in][Pytorch] 调整多模态模型存放目录 Merge pull request !6719 from zhangjunyi08/master	1 年前
inception.py	!6719 [built-in][Pytorch] 调整多模态模型存放目录 Merge pull request !6719 from zhangjunyi08/master	1 年前
metrics.py	!6719 [built-in][Pytorch] 调整多模态模型存放目录 Merge pull request !6719 from zhangjunyi08/master	1 年前

DDIM Sampling (Faster)

python -m torch.distributed.launch --nproc_per_node=8 --master_port 48109  utils/gen.py \ 
        --exp_name="pidm_deepfashion" \
        --checkpoint_name=last.pt \
        --dataset_path "./dataset/deepfashion/"  \
        --sample_algorithm ddim

DDPM Sampling

python -m torch.distributed.launch --nproc_per_node=8 --master_port 48109  utils/gen.py \ 
        --exp_name="pidm_deepfashion" \
        --checkpoint_name=last.pt \
        --dataset_path "./dataset/deepfashion/"  \
        --sample_algorithm ddpm

Output images are saved inside images folder.

Folder structure for checkpoint files

PIDM/
  checkpoints/
    <exp_name>/
      <checkpoint_name.pt>