Minor fix to both plot files

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

Minor fix to both plot files #4

Closed mahenning closed 4 years ago

mahenning commented 4 years ago

Only thing I'm not sure about is the change in tf.train.summary_iterator. IDE says it cant find the method and the one I included was the closest one I could find. Maybe you could verify that that's the right method to use.

Also had to exclude the gitignore file from parsing as it crashes at the groups line.

maximilianigl commented 4 years ago

The change to summary_iterator should be fine, at least for my tf version it points to the same function. If you're planning on running the Coinrun environment as well, I'd recommend creating a virtualenv with a TF1 version. Thank you for your PRs!