cnulab/RealNet

checkpoint

Opened this issue · 2 comments

作者您好,用自己的数据集微调diffusion模型生成的权重文件是.pth.tar而不是.pt后续合成异常数据时该如何修改
image
image

直接修改sample.yaml文件中的模型权重文件后缀就可以了。
https://github.com/cnulab/RealNet/blob/main/experiments/MVTec-AD/sample.yaml#L25

直接修改sample.yaml文件中的模型权重文件后缀就可以了。 https://github.com/cnulab/RealNet/blob/main/experiments/MVTec-AD/sample.yaml#L25

作者您好,为什么一次训练会生成多个ckpt文件呢?我训练结束之后如果想进行样本合成,需要以哪一个ckpt为准?