How to Implement Out-of-Distribution Detection on the MNIST Dataset?

ganmodokix / gwae

PyTorch Implementation for Gromov-Wasserstein Autoencoders (GWAE)

MIT License

42 stars 4 forks source link

How to Implement Out-of-Distribution Detection on the MNIST Dataset? #2

Open linyaosuna opened 3 months ago

linyaosuna commented 3 months ago

I am interested in implementing out-of-distribution detection on the mnist dataset, but I want to know what command should I use to achieve it. Looking forward to your reply！

ganmodokix commented 3 months ago

@linyaosuna Thank you for your interest! To implement out-of-distribution detection on the MNIST dataset, you can use the following command:

python eval.py "(path to .pth file trained on MNIST)"

For more detailed instructions, please use the --help option. If you have any further questions, feel free to ask!

linyaosuna commented 3 months ago

Thank you for your patient reply! It's really helpful to me!