How to reproduce mambaad on ader

lewandofskee / MambaAD

[NeurIPS 2024] Official implementation of MambaAD: Exploring State Space Models for Multi-class Unsupervised Anomaly Detection.

122 stars 3 forks source link

How to reproduce mambaad on ader #9

Closed jcjlin closed 4 weeks ago

jcjlin commented 1 month ago

Hi! Thank you for providing this amazing project! I have successfully cloned the repository and set up the necessary dependencies and dataset (MVTec). However, I am a bit unsure about how to proceed with reproducing the results. Could you kindly provide more detailed steps for running the model after the setup?

lewandofskee commented 1 month ago

First, you need to copy the contents of the mambaad repository to ader, MambaAD and ADer should be the same level of directory, for example, after copying the directory should be ader/configs/mambaad

Next, You can then run the following command to start the training:

CUDA_VISIBLE_DEVICES=0 python3 run.py -c configs/mambaad/mambaad_mvtec.py -m train