cnulab/RealNet

生成強度調整問題

RDADADA opened this issue · 7 comments

RDADADA commented

作者大大您好，

  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，請問是要將其中的檔案改名嗎?

2024-05-02 232013

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

感謝作者大大

cnulab commented

您好！

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

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

cnulab commented

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

  • 👍1
RDADADA commented

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

cnulab commented

使用

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

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

  • 👍1
RDADADA commented

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

cnulab commented

😄😄