生成強度調整問題

[RDADADA]

作者大大您好，

1. 請問生成條度調整這個執行碼 $ python -m torch.distributed.launch --nproc_per_node=1 sample.py --dataset MVTec-AD，
   所用到的sample.yaml中diffusion 與classifier 的checkpoints分別為可以從您提供的連結下載，但下載下來並沒有mvtec_diffusion_ckpt_epoch_240.pt與mvtec_classifier_ckpt_epoch_428.pt，請問是要將其中的檔案改名嗎?

2. 另外想請問論文中有提到改變S可以調整生成影像強度，請問實際是上調整哪個參數呢?
   

感謝作者大大

[cnulab]

您好！

1. 我看你下载的mvtec_diffusion_ckpt_epoch_240.pt模型大小不对，正常大小应该是2G，下载完成以后不用改名，直接放在对应路径下即可，请您使用我提供的链接再下载一次试试。
2. 异常强度s在 sample.py 中的117行修改。

[RDADADA]

您好，我是在這個連結下載的，下載後是一個壓縮檔，您的意思是直接將壓縮檔放在該路徑下嗎?
解壓縮後並沒有mvtec_diffusion_ckpt_epoch_240.pt與mvtec_classifier_ckpt_epoch_428.pt

[cnulab]

您好！下载后不需要解压，直接将pt文件放到对应的路径下。

[RDADADA]

感謝作者大大，pt文件解決了，
另外想請問有參數可以指定使用兩顆GPU嗎，我的硬體是兩顆3060 12G的GPU

[cnulab]

使用

$ python -m torch.distributed.launch --nproc_per_node=2 sample.py --dataset MVTec-AD

另外修改sample.yaml中dataset中的batch_size，12GB的GPU设置batch_size=6可能比较合适。

[RDADADA]

謝謝作者指教，我需要將batch_size=4才能運行，您真的很強

[cnulab]

😄😄