Make this package pip installable

danielkelshaw / ConcreteDropout

PyTorch implementation of 'Concrete Dropout'

https://arxiv.org/abs/1705.07832

MIT License

14 stars 2 forks source link

Make this package pip installable #15

Closed arose13 closed 1 year ago

arose13 commented 1 year ago

I added a setup.py file so this package can be installed easily with a pip install git+<https-repo-address>

arose13 commented 1 year ago

It's a good idea in my opinion, I'll make the change with my next PR.

Quick question: It looks like their is no option to have dropout run at inference time but the paper shows plots where the ConcreteDropout has MC samples turned on. Do you mind if I add that feature if it isn't already present?

danielkelshaw commented 1 year ago

Please feel free