1 Star 0 Fork 1

刘智勇/gitee_classifier_free_diffusion

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
makefile 467 Bytes
一键复制 编辑 原始数据 按行查看 历史
asker9527 提交于 2023-04-17 16:25 +08:00 . “提交项目”
.PHONY : train
train:
CUDA_VISIBLE_DEVICES=4,5,6,7 torchrun --standalone --nproc_per_node=gpu train.py
.PHONY : sample
samplepict:
CUDA_VISIBLE_DEVICES=4,5,6,7 torchrun --standalone --nproc_per_node=gpu sample.py --ddim True --select quadratic --genbatch 80 --w 0.5
.PHONY : samplenpz
samplenpz:
CUDA_VISIBLE_DEVICES=4,5,6,7 torchrun --standalone --nproc_per_node=gpu sample.py --fid True
.PHONY : clean
clean:
rm -rf __pycache__
rm -rf model/*
rm -rf sample/*
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Python
1
https://gitee.com/liu-zhiyong-wx/gitee_classifier_free_diffusion.git
git@gitee.com:liu-zhiyong-wx/gitee_classifier_free_diffusion.git
liu-zhiyong-wx
gitee_classifier_free_diffusion
gitee_classifier_free_diffusion
master

搜索帮助