search

gtegner / mine-pytorch

Mutual Information Neural Estimation in Pytorch
MIT License
283 stars 54 forks source link

Please share requirements #2

Closed aleXiehta closed 2 years ago

aleXiehta commented 3 years ago

Hi, MI estimation is really fascinating topic, and I'm glad that there's finally an implementation of all experiments in MINE.

However, I need the system requirements to run your codes. Would you mind to share the requirements with me? (Mainly about pytorch lightning

Thanks a lot!

jjanicechen commented 3 years ago

Hi @aleXiehta , try pytorch-lightning 0.6.0 ?

aleXiehta commented 2 years ago

Hi @aleXiehta , try pytorch-lightning 0.6.0 ?

All works well! Thank you for your help!