Fixed imports and other errors in experiments.py, also added a --cuda…

microsoft / IBAC-SNI

Code to reproduce the NeurIPS 2019 paper "Generalization in Reinforcement Learning with Selective Noise Injection and Information Bottleneck" by Maximilian Igl, Kamil Ciosek, Yingzhen Li, Sebastian Tschiatschek, Cheng Zhang, Sam Devlin and Katja Hofmann.

https://arxiv.org/abs/1910.12911

Other

52 stars 17 forks source link

Fixed imports and other errors in experiments.py, also added a --cuda… #3

Closed mahenning closed 4 years ago

mahenning commented 4 years ago

… parser option.

Hi, I could not reproduce the toy classification experiment with the given python commands in your Readme. I also added an empty results folder, otherwise the script would crash as it would try to write a file there the first time. This is my first ever pull request so if there's something missing or wrong please tell me!

msftclas commented 4 years ago

All CLA requirements met.

maximilianigl commented 4 years ago

Thank you! Does the code work for you now or is there something still going wrong?

mahenning commented 4 years ago

I added another pull request, I didn't realise that the gitignore file messes up the plot file.