search

cnulab / RealNet

Offical implementation of "RealNet: A Feature Selection Network with Realistic Synthetic Anomaly for Anomaly Detection (CVPR 2024)"
MIT License
143 stars 5 forks source link

在只有一个GPU资源下可以训练吗？ #41

Open wwj723452017 opened 4 days ago

wwj723452017 commented 4 days ago

作者你好，readme里面写的是“We use 4A40 GPUs and take 48 hours training the diffusion model, and use 2RTX3090 GPUs and take 3 hours training the guided classifier.” 想请问在只有一个GPU下可以训练模型吗，是否通过修改超参数实现?

cnulab commented 4 days ago

你好！ 只有一个GPU扩散模型可能训练不了，后面的步骤比如采样以及训练RealNet都是可以的。你如果只在比如MVTec或者VisA这样的数据集上实验，可以直接下载我提供的模型（diffusion和guided classifier）。